2121
2222def _df_to_schema (class_name , df , np , embedded = None , id_col = None , na_mode = None , keys = None ):
2323 """Convert a pandas DataFrame to a TerminusDB schema class definition.
24-
24+
2525 Args:
2626 class_name: Name of the schema class to create
2727 df: pandas DataFrame with columns to convert
@@ -30,23 +30,23 @@ def _df_to_schema(class_name, df, np, embedded=None, id_col=None, na_mode=None,
3030 id_col: Column name to use as document ID
3131 na_mode: NA handling mode ('error', 'skip', or 'optional')
3232 keys: List of column names to use as keys
33-
33+
3434 Returns:
3535 dict: Schema class definition dictionary
3636 """
3737 if keys is None :
3838 keys = []
3939 if embedded is None :
4040 embedded = []
41-
41+
4242 class_dict = {"@type" : "Class" , "@id" : class_name }
4343 np_to_builtin = {
4444 v : getattr (builtins , k )
4545 for k , v in np .sctypeDict .items ()
4646 if k in vars (builtins )
4747 }
4848 np_to_builtin [np .datetime64 ] = dt .datetime
49-
49+
5050 for col , dtype in dict (df .dtypes ).items ():
5151 if embedded and col in embedded :
5252 converted_type = class_name
@@ -62,7 +62,7 @@ def _df_to_schema(class_name, df, np, embedded=None, id_col=None, na_mode=None,
6262 class_dict [col ] = {"@type" : "Optional" , "@class" : converted_type }
6363 else :
6464 class_dict [col ] = converted_type
65-
65+
6666 return class_dict
6767
6868
0 commit comments