Update client.py

Author: raphaeljafriLB

labelpandas/client.py

@@ -138,6 +138,7 @@ def create_data_rows_from_table(
             client=self.lb_client, table=table, table_dict=table_dict, 
             row_data_col=x["row_data_col"], global_key_col=x["global_key_col"], external_id_col=x["external_id_col"], 
             dataset_id_col=x["dataset_id_col"], dataset_id=dataset_id, project_id_col=x["project_id_col"], project_id=project_id,
+            model_id_col=x["model_id_col"], model_id=model_id, model_run_id_col=x["model_run_id_col"], model_run_id=model_run_id,
             metadata_index=x["metadata_index"], attachment_index=x["attachment_index"], annotation_index=x["annotation_index"],
             create_action=actions["create"], annotate_action=actions["annotate"], prediction_action=actions["predictions"],
             upload_method=upload_method, mask_method=mask_method, divider=divider, verbose=verbose
@@ -277,13 +278,14 @@ def upsert_data_rows_from_table(batch_data_rows:bool=False, anootation_method:st
         # }
         # This uniforms the upload to use labelbase - Labelbox base code for best practices
         upload_dict = uploader.create_upload_dict( # Using labelpandas.uploader.create_upload_dict
-            client=self.lb_client, table=table, table_dict=table_dict, row_data_col=x["row_data_col"], global_key_col=x["global_key_col"], 
-            external_id_col=x["external_id_col"], dataset_id_col=x["dataset_id_col"], dataset_id=dataset_id, 
-            project_id_col=x["project_id_col"], project_id=project_id, metadata_index=x["metadata_index"], 
-            attachment_index=x["attachment_index"], annotation_index=x["annotation_index"],prediction_index=x["prediction_index"],
+            client=self.lb_client, table=table, table_dict=table_dict, 
+            row_data_col=x["row_data_col"], global_key_col=x["global_key_col"], external_id_col=x["external_id_col"], 
+            dataset_id_col=x["dataset_id_col"], dataset_id=dataset_id, project_id_col=x["project_id_col"], project_id=project_id,
+            model_id_col=x["model_id_col"], model_id=model_id, model_run_id_col=x["model_run_id_col"], model_run_id=model_run_id,
+            metadata_index=x["metadata_index"], attachment_index=x["attachment_index"], annotation_index=x["annotation_index"],
             create_action=actions["create"], annotate_action=actions["annotate"], prediction_action=actions["predictions"],
             upload_method=upload_method, mask_method=mask_method, divider=divider, verbose=verbose
-        )     
+        )
 
         # Bath to project attempt using labelbase.uploader.batch_rows_to_project
         if actions["batch"] and batch_data_rows: