1919from ..woqlschema .woql_schema import WOQLSchema
2020
2121
22- def _df_to_schema (class_name , df , np , embedded = None , id_col = None , na_mode = None , keys = None ):
22+ def _df_to_schema (
23+ class_name , df , np , embedded = None , id_col = None , na_mode = None , keys = None
24+ ):
2325 """Convert a pandas DataFrame to a TerminusDB schema class definition.
2426
2527 Args:
@@ -41,9 +43,7 @@ def _df_to_schema(class_name, df, np, embedded=None, id_col=None, na_mode=None,
4143
4244 class_dict = {"@type" : "Class" , "@id" : class_name }
4345 np_to_builtin = {
44- v : getattr (builtins , k )
45- for k , v in np .sctypeDict .items ()
46- if k in vars (builtins )
46+ v : getattr (builtins , k ) for k , v in np .sctypeDict .items () if k in vars (builtins )
4747 }
4848 np_to_builtin [np .datetime64 ] = dt .datetime
4949
@@ -512,7 +512,15 @@ def importcsv(
512512 converted_col = col .lower ().replace (" " , "_" ).replace ("." , "_" )
513513 df .rename (columns = {col : converted_col }, inplace = True )
514514 if not has_schema :
515- class_dict = _df_to_schema (class_name , df , np , embedded = embedded , id_col = id_ , na_mode = na , keys = keys )
515+ class_dict = _df_to_schema (
516+ class_name ,
517+ df ,
518+ np ,
519+ embedded = embedded ,
520+ id_col = id_ ,
521+ na_mode = na ,
522+ keys = keys ,
523+ )
516524 if message is None :
517525 schema_msg = f"Schema object insert/ update with { csv_file } by Python client."
518526 else :
0 commit comments