File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change 4545 UseCase ,
4646 UseCaseMetadata ,
4747)
48+ from api .utils .enums import MetadataModels
4849
4950
5051def parse_geography_value (value : str ) -> List [str ]:
@@ -76,7 +77,9 @@ def migrate_dataset_geographies(
7677
7778 # Get geography metadata item
7879 try :
79- geo_metadata_item = Metadata .objects .get (label = "Geography" )
80+ geo_metadata_item = Metadata .objects .get (
81+ label = "Geography" , model = MetadataModels .DATASET
82+ )
8083 except Metadata .DoesNotExist :
8184 print ("⚠️ No Geography metadata item found. Skipping dataset migration." )
8285 return
@@ -169,7 +172,9 @@ def migrate_usecase_geographies(
169172
170173 # Get geography metadata item
171174 try :
172- geo_metadata_item = Metadata .objects .get (label = "Geography" )
175+ geo_metadata_item = Metadata .objects .get (
176+ label = "Geography" , model = MetadataModels .USECASE
177+ )
173178 except Metadata .DoesNotExist :
174179 print ("⚠️ No Geography metadata item found. Skipping usecase migration." )
175180 return
You can’t perform that action at this time.
0 commit comments