-- bugfix for integer column pandas dataframes

manujosephv · manujosephv · commit 132c3f29e46e · 2021-06-19T17:16:54.000+05:30
diff --git a/pytorch_tabular/config/config.py b/pytorch_tabular/config/config.py
@@ -81,7 +81,7 @@ class DataConfig:
         default=MISSING,
         metadata={"help": "A list of strings with the names of the target column(s)"},
     )
-    continuous_cols: List[str] = field(
+    continuous_cols: List = field(
         default_factory=list,
         metadata={"help": "Column names of the numeric fields. Defaults to []"},
     )

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ class DataConfig:`
`81`	`81`	`default=MISSING,`
`82`	`82`	`metadata={"help": "A list of strings with the names of the target column(s)"},`
`83`	`83`	`)`
`84`		`- continuous_cols: List[str] = field(`
	`84`	`+ continuous_cols: List = field(`
`85`	`85`	`default_factory=list,`
`86`	`86`	`metadata={"help": "Column names of the numeric fields. Defaults to []"},`
`87`	`87`	`)`