From d7aef66a07d90b2c6d080be7cd2011af467a2e96 Mon Sep 17 00:00:00 2001
From: Tabassum Kakar <tabassum_kakar@hms.harvard.edu>
Date: Fri, 25 Oct 2024 14:10:41 -0400
Subject: [PATCH] Fixed test-fixture

---
 src/portal_visualization/builder_factory.py            | 10 +++++-----
 .../d58d16d515fe3237b0ca793da68a5d48.json              |  3 +--
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/portal_visualization/builder_factory.py b/src/portal_visualization/builder_factory.py
index e9ff898..51964ff 100644
--- 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
index 220c290..9bba437 100644
--- 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"
     ]
   }
\ No newline at end of file