last pkg_resources

Signed-off-by: Danny C <d.chiao@gmail.com>
feast-dev · Sep 7, 2023 · 9af1cad · 9af1cad
1 parent 3a40c07
commit 9af1cad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sdk/python/feast/proto_json.py b/sdk/python/feast/proto_json.py
@@ -118,7 +118,7 @@ def from_json_object_updated(
 
     # https://github.com/feast-dev/feast/issues/2484 Certain feast users need a higher version of protobuf but the
     # parameters of `from_json_object` changes in feast 3.20.1. This change gives users flexibility to use earlier versions.
-    current_version = importlib.resources.get_distribution("protobuf").version
+    current_version = importlib_version("protobuf")
     if version.parse(current_version) < version.parse("3.20"):
         _patch_proto_json_encoding(Value, to_json_object, from_json_object)
     else:
@@ -221,7 +221,7 @@ def from_json_object_updated(
 
     # https://github.com/feast-dev/feast/issues/2484 Certain feast users need a higher version of protobuf but the
     # parameters of `from_json_object` changes in feast 3.20.1. This change gives users flexibility to use earlier versions.
-    current_version = pkg_resources.get_distribution("protobuf").version
+    current_version = importlib_version("protobuf")
     if version.parse(current_version) < version.parse("3.20"):
         _patch_proto_json_encoding(FeatureList, to_json_object, from_json_object)
     else: