@@ -226,16 +226,15 @@ def _reorder_new_dataset(
226226def _generate_primary_key (
227227 dataset : pd .DataFrame , primary_key_cols : List [str ], primary_key : str
228228) -> pd .DataFrame :
229- """
230- Generate primary key column a dataframe
229+ """Generate primary key column a dataframe
231230
232231 Args:
233- dataset(pd.DataFrame): A dataframe
234- new_dataset: The re-ordered new dataset
235- primary_key_cols (list ): Column(s) that make up the primary key
236- primary_key: The column name of the primary_key
232+ dataset (pd.DataFrame): A dataframe
233+ primary_key_cols (List[str]): Column(s) that make up the primary key
234+ primary_key (str ): The column name of the primary_key
235+
237236 Returns:
238- The dataframe with primary_key column added
237+ pd.DataFrame: The dataframe with primary_key column added
239238 """
240239 # replace NAs with emtpy string
241240 dataset = dataset .fillna ("" )
@@ -298,21 +297,18 @@ def store_database(
298297 syn : synapseclient .Synapse ,
299298 database_synid : str ,
300299 col_order : List [str ],
301- allupdates : pd .DataFrame ,
300+ all_updates : pd .DataFrame ,
302301 to_delete_rows : pd .DataFrame ,
303- ):
302+ ) -> None :
304303 """
305304 Store changes to the database
306305
307306 Args:
308- syn (synapseclient.Synaps ): Synapse object
307+ syn (synapseclient.Synapse ): Synapse object
309308 database_synid (str): Synapse Id of the Synapse table
310309 col_order (List[str]): The ordered column names to be saved
311- allupdates (pd.DataFrame): rows to be appended and/or updated
310+ all_updates (pd.DataFrame): rows to be appended and/or updated
312311 to_deleted_rows (pd.DataFrame): rows to be deleted
313-
314- Returns:
315- None
316312 """
317313 storedatabase = False
318314 update_all_file = tempfile .NamedTemporaryFile (
@@ -321,14 +317,14 @@ def store_database(
321317 with open (update_all_file .name , "w" ) as updatefile :
322318 # Must write out the headers in case there are no appends or updates
323319 updatefile .write ("," .join (col_order ) + "\n " )
324- if not allupdates .empty :
320+ if not all_updates .empty :
325321 """
326322 This is done because of pandas typing.
327323 An integer column with one NA/blank value
328324 will be cast as a double.
329325 """
330326 updatefile .write (
331- allupdates [col_order ]
327+ all_updates [col_order ]
332328 .to_csv (index = False , header = None )
333329 .replace (".0," , "," )
334330 .replace (".0\n " , "\n " )
0 commit comments