Merge pull request #1734 from Giskard-AI/GSK-2468

GSK-2468 Try to automatically add import when adding kwargs tests to giskard Hub

giskard/core/kwargs_utils.py

@@ -0,0 +1,30 @@
+from typing import Dict, List, Optional, Set, Tuple, Union
+
+import inspect
+from collections import defaultdict
+
+
+def get_imports_code(obj):
+    imports = _get_imports(obj)
+    return "\n".join([f'from {module} import {", ".join(classes)}' for module, classes in imports.items()])
+
+
+def _get_imports(obj, imports: Optional[defaultdict] = None):
+    imports = imports or defaultdict(set)
+    module = inspect.getmodule(obj)
+
+    if isinstance(obj, Dict):
+        for i in obj.values():
+            imports = _get_imports(i, imports)
+    elif isinstance(obj, Union[List, Set, Tuple]):
+        for i in obj:
+            print(i)
+            imports = _get_imports(i, imports)
+    elif hasattr(obj, "__dict__"):
+        imports = _get_imports(obj.__dict__.values(), imports)
+
+    if module is None:
+        return imports
+
+    imports[module.__name__].add(obj.__class__.__name__)
+    return imports

giskard/core/suite.py

@@ -25,6 +25,7 @@
 from ..client.python_utils import warning
 from ..utils.analytics_collector import analytics
 from ..utils.artifacts import serialize_parameter
+from .kwargs_utils import get_imports_code
 
 logger = logging.getLogger(__name__)
 
@@ -514,7 +515,7 @@ def to_dto(self, client: GiskardClient, project_key: str, uploaded_uuid_status:
             )
 
             kwargs_params = [
-                f"kwargs[{repr(pname)}] = {repr(value)}"
+                f"{get_imports_code(value)}\nkwargs[{repr(pname)}] = {repr(value)}"
                 for pname, value in t.provided_inputs.items()
                 if pname not in t.giskard_test.meta.args
             ]