cover dicts which are not param-dicts

basf · Oct 28, 2024 · 43b39e9 · 43b39e9
1 parent 9466892
commit 43b39e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/molpipeline/utils/comparison.py b/molpipeline/utils/comparison.py
@@ -32,6 +32,8 @@ def remove_irrelevant_params(params: _T) -> _T:
     if isinstance(params, dict):
         params_new = {}
         for key, value in params.items():
+            if not isinstance(key, str):
+                continue
             if key.split("__")[-1] in irrelevant_params:
                 continue
             params_new[key] = remove_irrelevant_params(value)