Fix imports in test_types

mckinsey · Nov 23, 2023 · 941eb8a · 941eb8a
1 parent 452657e
commit 941eb8a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/vizro-core/tests/unit/vizro/models/test_types.py b/vizro-core/tests/unit/vizro/models/test_types.py
@@ -1,3 +1,6 @@
+import importlib
+from pathlib import Path
+
 import plotly.express as plotly_express
 import plotly.graph_objects as go
 import pytest
@@ -145,7 +148,7 @@ def invalid_decorated_graph_function():
 
 class Model(VizroBaseModel):
     # The import_path here makes it possible to import the above function using getattr(import_path, _target_).
-    function: CapturedCallable = Field(..., import_path=__import__(__name__))
+    function: CapturedCallable = Field(..., import_path=importlib.import_module(__name__))
 
 
 class TestModelFieldPython: