Fix import spyt.types

1f6aac7347755f1708d97732fe38406faa6622e3
ytsaurus · Apr 17, 2024 · 1b14f41 · 1b14f41
1 parent 51c87ad
commit 1b14f41
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yt/spark/spark-over-yt/spyt-package/src/main/python/spyt/__init__.py b/yt/spark/spark-over-yt/spyt-package/src/main/python/spyt/__init__.py
@@ -13,7 +13,7 @@
 import pyspark.sql.readwriter
 import pyspark.cloudpickle.cloudpickle
 import pyspark.cloudpickle.cloudpickle_fast
-import types
+from types import CodeType
 
 __all__ = [
     'connect',
@@ -47,7 +47,7 @@ def initialize():
     pyspark.cloudpickle.cloudpickle._extract_code_globals = _extract_code_globals
     pyspark.cloudpickle.cloudpickle_fast._extract_code_globals = _extract_code_globals
     pyspark.cloudpickle.cloudpickle_fast._code_reduce = _code_reduce
-    pyspark.cloudpickle.cloudpickle_fast.CloudPickler.dispatch_table[types.CodeType] = _code_reduce
+    pyspark.cloudpickle.cloudpickle_fast.CloudPickler.dispatch_table[CodeType] = _code_reduce
 
 
 initialize()