Update requirements

keras-team · Dec 12, 2023 · ee46587 · ee46587
1 parent cd0cbc4
commit ee46587
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/examples/vision/video_classification.py b/examples/vision/video_classification.py
@@ -218,7 +218,8 @@ def prepare_all_videos(df, root_dir):
             length = min(MAX_SEQ_LENGTH, video_length)
             for j in range(length):
                 temp_frame_features[i, j, :] = feature_extractor.predict(
-                    batch[None, j, :], verbose=0,
+                    batch[None, j, :],
+                    verbose=0,
                 )
             temp_frame_mask[i, :length] = 1  # 1 = not masked, 0 = masked
 

diff --git a/requirements.txt b/requirements.txt
@@ -10,7 +10,7 @@ pandas
 jupyter
 pycocotools
 pydot
-tensorflow==2.15.0
+tensorflow
 tensorflow_datasets
 keras-tuner==1.4.5
 keras-cv==0.7.1