[r] Use DCP2 schema name for HCA replica_type (#6453)

DataBiosphere · Jul 31, 2024 · 17ef083 · 17ef083
1 parent d4dac36
commit 17ef083
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/azul/plugins/metadata/hca/indexer/transform.py b/src/azul/plugins/metadata/hca/indexer/transform.py
@@ -462,8 +462,8 @@ class BaseTransformer(Transformer, metaclass=ABCMeta):
     api_bundle: api.Bundle
 
     def replica_type(self, entity: EntityReference) -> str:
-        assert entity.entity_type == self.entity_type(), entity
-        return entity.entity_type.removesuffix('s')
+        api_entity = self.api_bundle.entities[UUID(entity.entity_id)]
+        return api_entity.schema_name
 
     @classmethod
     def aggregator(cls, entity_type: EntityType) -> Optional[EntityAggregator]: