File tree Expand file tree Collapse file tree 1 file changed +10
-8
lines changed
Expand file tree Collapse file tree 1 file changed +10
-8
lines changed Original file line number Diff line number Diff line change @@ -197,8 +197,8 @@ def run_direct_relational_import(
197197 # Check if the field exists in the DataFrame
198198 if field not in source_df .columns :
199199 log .error (
200- f"Field '{ field } ' not found in source DataFrame. "
201- f"Available columns: { source_df .columns } "
200+ f"Field '{ field } ' not found in source DataFrame. "
201+ f"Available columns: { source_df .columns } "
202202 )
203203 return None
204204
@@ -277,12 +277,14 @@ def _prepare_link_dataframe(
277277 f"Available columns: { source_df .columns } "
278278 )
279279 # Return an empty DataFrame with the expected schema
280- return pl .DataFrame (schema = {
281- "external_id" : pl .Utf8 ,
282- field : pl .Utf8 ,
283- owning_model_fk : pl .Int64 ,
284- f"{ related_model_fk } /id" : pl .Int64
285- })
280+ return pl .DataFrame (
281+ schema = {
282+ "external_id" : pl .Utf8 ,
283+ field : pl .Utf8 ,
284+ owning_model_fk : pl .Int64 ,
285+ f"{ related_model_fk } /id" : pl .Int64 ,
286+ }
287+ )
286288
287289 # Create the link table DataFrame
288290 link_df = source_df .select (["id" , field ]).rename ({"id" : "external_id" })
You can’t perform that action at this time.
0 commit comments