Update IExplainTask signature

openvinotoolkit · Mar 27, 2023 · 3fd75ae · 3fd75ae
1 parent e5e254b
commit 3fd75ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/otx/api/usecases/tasks/interfaces/explain_interface.py b/otx/api/usecases/tasks/interfaces/explain_interface.py
@@ -8,7 +8,7 @@
 import abc
 
 from otx.api.entities.datasets import DatasetEntity
-from otx.api.entities.inference_parameters import InferenceParameters
+from otx.api.entities.explain_parameters import ExplainParameters
 
 
 class IExplainTask(metaclass=abc.ABCMeta):
@@ -18,7 +18,7 @@ class IExplainTask(metaclass=abc.ABCMeta):
     def explain(
         self,
         dataset: DatasetEntity,
-        explain_parameters: InferenceParameters,
+        explain_parameters: ExplainParameters,
     ) -> DatasetEntity:
         """This is the method that is called upon explanation.