Address comments

Signed-off-by: Iaroslav Igoshev <[email protected]>

modin/core/storage_formats/pandas/query_compiler.py

@@ -1976,7 +1976,10 @@ def astype(self, col_dtypes, errors: str = "raise"):
         # other query compilers may not take care of error handling at the API
         # layer. This query compiler assumes there won't be any errors due to
         # invalid type keys.
-        return self.__constructor__(self._modin_frame.astype(col_dtypes, errors=errors))
+        return self.__constructor__(
+            self._modin_frame.astype(col_dtypes, errors=errors),
+            shape_hint=self._shape_hint,
+        )
 
     def infer_objects(self):
         return self.__constructor__(self._modin_frame.infer_objects())

modin/pandas/series.py

@@ -149,7 +149,6 @@ def __init__(
                 )
             )._query_compiler
         self._query_compiler = query_compiler.columnarize()
-        self._query_compiler._shape_hint = "column"
         if name is not None:
             self.name = name