minor typing annotation

ITISFoundation · Mar 23, 2022 · 8b1a9c3 · 8b1a9c3
1 parent 3304703
commit 8b1a9c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/packages/models-library/tests/test_projects_pipeline.py b/packages/models-library/tests/test_projects_pipeline.py
@@ -3,7 +3,7 @@
 # pylint:disable=redefined-outer-name
 
 from pprint import pformat
-from typing import Dict
+from typing import Dict, Type
 
 import pytest
 from models_library.projects_pipeline import ComputationTask
@@ -15,7 +15,7 @@
     (ComputationTask,),
 )
 def test_computation_task_model_examples(
-    model_cls: BaseModel, model_cls_examples: Dict[str, Dict]
+    model_cls: Type[BaseModel], model_cls_examples: Dict[str, Dict]
 ):
     for name, example in model_cls_examples.items():
         print(name, ":", pformat(example))