Correct subtype in mapping

ministryofjustice · Dec 3, 2024 · df3cc43 · df3cc43
1 parent e2ff430
commit df3cc43
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/datahub-client/data_platform_catalogue/client/search.py b/lib/datahub-client/data_platform_catalogue/client/search.py
@@ -61,7 +61,7 @@ def __init__(self, graph: DataHubGraph):
             (DatahubEntityType.DATASET.value, DatahubSubtype.SOURCE.value): (self._parse_dataset, EntityTypes.TABLE),
             (DatahubEntityType.CONTAINER.value, DatahubSubtype.DATABASE.value): (self._parse_container, EntityTypes.DATABASE),
             (DatahubEntityType.CONTAINER.value, DatahubSubtype.PUBLICATION_COLLECTION.value): (self._parse_container, EntityTypes.PUBLICATION_COLLECTION),
-            (DatahubEntityType.DATASET.value, DatahubSubtype.PUBLICATION_DATASET.value): (self._parse_container, EntityTypes.PUBLICATION_COLLECTION),
+            (DatahubEntityType.DATASET.value, DatahubSubtype.PUBLICATION_DATASET.value): (self._parse_container, EntityTypes.PUBLICATION_DATASET),
             (DatahubEntityType.CHART.value, None): (self._parse_dataset, EntityTypes.CHART),
             (DatahubEntityType.DASHBOARD.value, None): (self._parse_container, EntityTypes.DASHBOARD),
         }