Remove str conversions

sdv-dev · May 3, 2024 · 8982a70 · 8982a70
1 parent 8d617bd
commit 8982a70
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/sdv/metadata/single_table.py b/sdv/metadata/single_table.py
@@ -968,10 +968,9 @@ def validate(self):
                 + '\n'.join([str(e) for e in errors])
             )
 
-    def _validate_metadata_matches_data(self, data_columns):
+    def _validate_metadata_matches_data(self, columns):
         errors = []
         metadata_columns = self.columns or {}
-        columns = data_columns.astype(str)
         missing_data_columns = set(columns).difference(metadata_columns)
         if missing_data_columns:
             errors.append(
@@ -1060,7 +1059,7 @@ def _validate_column_data(self, column, sdtype_warnings):
             list:
                 A list containing any validation error messages found during the process.
         """
-        column_metadata = self.columns[str(column.name)]
+        column_metadata = self.columns[column.name]
         sdtype = column_metadata['sdtype']
         invalid_values = None