PR comments and tests

modin/core/storage_formats/pandas/query_compiler_validator.py

@@ -20,6 +20,7 @@
 """
 import functools
 import inspect
+from pandas.core.indexes.frozen import FrozenList
 from types import FunctionType, MethodType
 from typing import Any, Dict, Tuple, TypeVar
 
@@ -56,10 +57,13 @@ def cast_arg_to_current_qc(arg):
         else:
             return arg
 
-    if isinstance(arguments, tuple):
+    imutable_types = (FrozenList, tuple)
+    if isinstance(arguments, imutable_types):
+        args_type = type(arguments)
         arguments = list(arguments)
         arguments = cast_nested_args_to_current_qc_type(arguments, current_qc)
-        return tuple(arguments)
+
+        return args_type(arguments)
     if isinstance(arguments, list):
         for i in range(len(arguments)):
             if isinstance(arguments[i], (list, dict)):
@@ -88,8 +92,6 @@ def apply_argument_cast():
     def decorator(obj: Fn) -> Fn:
         """Decorate function or class to cast all arguments that are query compilers to the current query compiler"""
         if isinstance(obj, type):
-            seen: Dict[Any, Any] = {}
-
             all_attrs = dict(inspect.getmembers(obj))
             all_attrs.pop("__abstractmethods__")
 
@@ -103,10 +105,7 @@ def decorator(obj: Fn) -> Fn:
                 if isinstance(
                     attr_value, (FunctionType, MethodType, classmethod, staticmethod)
                 ):
-                    try:
-                        wrapped = seen[attr_value]
-                    except KeyError:
-                        wrapped = seen[attr_value] = apply_argument_cast()(attr_value)
+                    wrapped = apply_argument_cast()(attr_value)
                     setattr(obj, attr_name, wrapped)
             return obj  # type: ignore [return-value]
         elif isinstance(obj, classmethod):