fix: ruff check

radicalbit · Jun 28, 2024 · 007989b · 007989b
1 parent 8e32bbc
commit 007989b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdk/radicalbit_platform_sdk/apis/model_current_dataset.py b/sdk/radicalbit_platform_sdk/apis/model_current_dataset.py
@@ -11,6 +11,7 @@
     BinaryClassificationDataQuality,
     CurrentBinaryClassificationModelQuality,
     CurrentFileUpload,
+    CurrentMultiClassificationModelQuality,
     DataQuality,
     DatasetStats,
     Drift,
@@ -23,9 +24,6 @@
     RegressionDrift,
     RegressionModelQuality,
 )
-from radicalbit_platform_sdk.models.dataset_model_quality import (
-    CurrentMultiClassificationModelQuality,
-)
 
 
 class ModelCurrentDataset:

diff --git a/sdk/radicalbit_platform_sdk/models/__init__.py b/sdk/radicalbit_platform_sdk/models/__init__.py
@@ -29,6 +29,7 @@
     CurrentBinaryClassificationModelQuality,
     ModelQuality,
     MultiClassificationModelQuality,
+    CurrentMultiClassificationModelQuality,
     RegressionModelQuality,
 )
 from .dataset_stats import DatasetStats
@@ -56,6 +57,7 @@
     'BinaryClassificationModelQuality',
     'CurrentBinaryClassificationModelQuality',
     'MultiClassificationModelQuality',
+    'CurrentMultiClassificationModelQuality',
     'RegressionModelQuality',
     'DataQuality',
     'BinaryClassificationDataQuality',