.loc fix

huggingface · Apr 23, 2024 · 2961150 · 2961150
1 parent e8ab0e6
commit 2961150
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -41,4 +41,4 @@
 
 
 logger = Logger().get_logger()
-__version__ = "0.7.57.dev0"
+__version__ = "0.7.58.dev0"
diff --git a/src/autotrain/preprocessor/text.py b/src/autotrain/preprocessor/text.py
@@ -80,8 +80,8 @@ def prepare(self):
         train_df, valid_df = self.split()
         train_df, valid_df = self.prepare_columns(train_df, valid_df)
 
-        train_df[:, "autotrain_label"] = train_df["autotrain_label"].astype(str)
-        valid_df[:, "autotrain_label"] = valid_df["autotrain_label"].astype(str)
+        train_df.loc[:, "autotrain_label"] = train_df["autotrain_label"].astype(str)
+        valid_df.loc[:, "autotrain_label"] = valid_df["autotrain_label"].astype(str)
 
         label_names = sorted(set(train_df["autotrain_label"].unique().tolist()))
Original file line number	Diff line number	Diff line change
Expand Up		@@ -41,4 +41,4 @@


		logger = Logger().get_logger()
		__version__ = "0.7.57.dev0"
		__version__ = "0.7.58.dev0"