Skip to content

Commit c1a6eaa

Browse files
MoFHekarhdong
authored andcommitted
[fix] With some pip versions if "tensorflow-recommenders-addons-gpu" is installed but running pkg_resources.get_distribution("tensorflow-recommenders-addons") will lead to the error with invalid version 'nsorflow-recommenders-addons-gpu'.
1 parent cd6e2b9 commit c1a6eaa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_recommenders_addons/utils/resource_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@
2626
def get_required_tf_version():
2727
try:
2828
pkg = pkg_resources.get_distribution("tensorflow-recommenders-addons")
29-
except pkg_resources.DistributionNotFound:
29+
except:
3030
try:
3131
pkg = pkg_resources.get_distribution("tensorflow-recommenders-addons-gpu")
3232
except pkg_resources.DistributionNotFound:

0 commit comments

Comments
 (0)