adds tf import

rnugmanx · Oct 25, 2023 · acc149c · acc149c
1 parent bef4225
commit acc149c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/e2e_oss/utils/test_utils.py b/tests/e2e_oss/utils/test_utils.py
@@ -8,7 +8,7 @@
 from typing import Union
 import copy
 import numpy as np
-
+import tensorflow as tf
 
 from filelock import FileLock