Remove astype

databricks · Dec 24, 2020 · d26b899 · d26b899
1 parent 7242716
commit d26b899
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/databricks/koalas/series.py b/databricks/koalas/series.py
@@ -2029,7 +2029,7 @@ def factorize(self, sort: bool = True, na_sentinel: Optional[int] = -1):
         mapped_scol = map_scol.getItem(self.spark.column)
         internal = self._internal.with_new_columns([null_scol.otherwise(mapped_scol)])
 
-        codes = first_series(DataFrame(internal)).astype(np.int64)
+        codes = first_series(DataFrame(internal))
 
         if na_sentinel is not None:
             uniques = ks.Index([x for x in uniques_list if x is not None])