Skip to content

Commit 8493136

Browse files
committed
Preventing re-initialization of Version
1 parent d1a3475 commit 8493136

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

tests/conftest.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -556,15 +556,18 @@ def tf_full_version(tensorflow_training_latest_version, tensorflow_inference_lat
556556
Fixture exists as such, since TF training and TFS have different latest versions.
557557
Otherwise, this would simply be a single latest version.
558558
"""
559-
if Version(tensorflow_training_latest_version) in SpecifierSet(">=2.16"):
559+
tensorflow_training_latest_version = Version(tensorflow_training_latest_version)
560+
tensorflow_inference_latest_version = Version(tensorflow_inference_latest_version)
561+
562+
if tensorflow_training_latest_version in SpecifierSet(">=2.16"):
560563
return (
561-
f"{Version(tensorflow_training_latest_version).major}"
562-
f".{Version(tensorflow_training_latest_version).minor}"
564+
f"{tensorflow_training_latest_version.major}"
565+
f".{tensorflow_training_latest_version.minor}"
563566
)
564567
return str(
565568
min(
566-
Version(tensorflow_training_latest_version),
567-
Version(tensorflow_inference_latest_version),
569+
tensorflow_training_latest_version,
570+
tensorflow_inference_latest_version,
568571
)
569572
)
570573

0 commit comments

Comments
 (0)