pipeline fix, hopefully - kamangir/bolt#746

kamangir · Jul 27, 2024 · d2f1fc0 · d2f1fc0
1 parent aa43bbc
commit d2f1fc0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/blue_geo/__init__.py b/blue_geo/__init__.py
@@ -4,7 +4,7 @@
 
 DESCRIPTION = f"{ICON} AI for precise geospatial data analysis and visualization."
 
-VERSION = "4.154.1"
+VERSION = "4.155.1"
 
 REPO_NAME = "blue-geo"
 

diff --git a/blue_geo/catalog/classes.py b/blue_geo/catalog/classes.py
@@ -33,5 +33,5 @@ def get_datacube_classes(
     return [
         datacube_class
         for datacube_class in list_of_datacube_classes
-        if isinstance(datacube_class.catalog, catalog_class)
+        if datacube_class.catalog.__class__ == catalog_class
     ]
diff --git a/blue_geo/tests/test_catalog.py b/blue_geo/tests/test_catalog.py
@@ -22,16 +22,16 @@ def test_list_of_catalog_classes(catalog_class: Type[GenericCatalog]):
 
 
 @pytest.mark.parametrize(
-    ["datacube_class", "datacube_id"],
+    ["datacube_id", "datacube_class"],
     [
         [datacube_id, datacube_class]
         for datacube_id, datacube_class in assets.datacubes.items()
         if datacube_class in list_of_datacube_classes
     ],
 )
 def test_list_of_datacube_classes(
-    datacube_class: Type[GenericDatacube],
     datacube_id: str,
+    datacube_class: Type[GenericDatacube],
 ):
     object_name = unique_object()
 
@@ -83,6 +83,6 @@ def test_get_datacube_classes(catalog_class: Type[GenericCatalog]):
     datacube_class_list = get_datacube_classes(catalog_class)
 
     assert all(
-        datacube_class.catalog == catalog_class
+        datacube_class.catalog.__class__ == catalog_class
         for datacube_class in datacube_class_list
     )