Fixed test-fixture

hubmapconsortium · Oct 25, 2024 · d7aef66 · d7aef66
1 parent 4823e6a
commit d7aef66
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/portal_visualization/builder_factory.py b/src/portal_visualization/builder_factory.py
@@ -34,7 +34,7 @@ def process_hints(hints):
     is_json = "json_based" in hints
     is_spatial = "spatial" in hints
     is_support = "is_support" in hints
-    is_segmentation_mask = "segmentation_mask" in hints
+    is_segmentation_base = "segmentation_base" in hints
 
     return (
         is_image,
@@ -46,7 +46,7 @@ def process_hints(hints):
         is_json,
         is_spatial,
         is_support,
-        is_segmentation_mask,
+        is_segmentation_base,
     )
 
 
@@ -72,13 +72,13 @@ def get_view_config_builder(entity, get_assaytype, parent=None, epic_uuid=None):
         is_json,
         is_spatial,
         is_support,
-        is_segmentation_mask
+        is_segmentation_base
     ) = process_hints(hints)
 
     # vis-lifted image pyramids
     if parent is not None:
         # TODO: For now epic (base image's) support datasets doesn't have any hints
-        if epic_uuid is not None:
+        if epic_uuid is not None or is_segmentation_base:
             return SegImagePyramidViewConfBuilder
 
         elif is_support and is_image:
@@ -142,5 +142,5 @@ def get_view_config_builder(entity, get_assaytype, parent=None, epic_uuid=None):
 
 
 def has_visualization(entity, get_assaytype, parent=None):
-    builder = get_view_config_builder(entity, get_assaytype, parent, None)
+    builder = get_view_config_builder(entity, get_assaytype, parent, epic_uuid=None)
     return builder != NullViewConfBuilder
diff --git a/test/assaytype-fixtures/d58d16d515fe3237b0ca793da68a5d48.json b/test/assaytype-fixtures/d58d16d515fe3237b0ca793da68a5d48.json
@@ -5,7 +5,6 @@
     "primary": false,
     "vitessce-hints": [
       "is_image",
-      "is_support",
-      "pyramid"
+      "segmentation_base"
     ]
   }