Get TF version from Python and not Pip (#498)

Co-authored-by: Ben Frederickson <ben@benfrederickson.com>
NVIDIA-Merlin · Aug 4, 2022 · cf20dc8 · cf20dc8
1 parent 5b1ff2f
commit cf20dc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/extractor.py b/docs/extractor.py
@@ -366,7 +366,7 @@ def main(args):
     xtr.get_from_python("hugectr2onnx")
     xtr.get_from_python("hugectr")
     xtr.get_from_python("sparse_operation_kit")
-    xtr.get_from_pip("tensorflow", "tf")
+    xtr.get_from_python("tensorflow", "tf")
     xtr.get_from_pip("torch", "pytorch")
     xtr.get_from_env("CUBLAS_VERSION", "cublas")
     xtr.get_from_env("CUFFT_VERSION", "cufft")