rm redundant code

innat · Mar 3, 2024 · af5878c · af5878c
1 parent 61303be
commit af5878c
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 9 deletions.
diff --git a/keras_cv/layers/video_swin_layers.py b/keras_cv/layers/video_swin_layers.py
@@ -97,7 +97,7 @@ def get_window_size(x_size, window_size, shift_size=None):
     <https://arxiv.org/abs/2103.14030>"
     https://github.com/microsoft/Swin-Transformer
 
-     Args:
+    Args:
         x_size: input size.
         window_size: local window size.
         shift_size: window shifting size.

diff --git a/keras_cv/models/backbones/video_swin/video_swin_backbone_test.py b/keras_cv/models/backbones/video_swin/video_swin_backbone_test.py
@@ -19,9 +19,12 @@
 
 from keras_cv.backend import keras
 from keras_cv.backend import ops
-from keras_cv.models.backbones.video_swin.video_swin_aliases import VideoSwinSBackbone
+from keras_cv.models.backbones.video_swin.video_swin_aliases import (
+    VideoSwinSBackbone,
+)
 from keras_cv.tests.test_case import TestCase
 
+
 class TestViTDetBackbone(TestCase):
     @pytest.mark.large
     def test_call(self):

diff --git a/keras_cv/models/classification/video_classifier_presets.py b/keras_cv/models/classification/video_classifier_presets.py
@@ -16,34 +16,34 @@
 classifier_presets = {
     "videoswin_tiny_kinetics_classifier": {
         "metadata": {
-            "description": ("videoswin_tiny_kinetics "),  # TODO: update
+            "description": ("videoswin_tiny_kinetics "),
             "params": 25_613_800,
             "official_name": "VideoClassifier",
             "path": "video_classifier",
         },
     },
     "videoswin_small_kinetics_classifier": {
         "metadata": {
-            "description": ("videoswin_small_kinetics "),  # TODO: update
-            "params": 25_613_800,  # TODO: update
+            "description": ("videoswin_small_kinetics "),
+            "params": 25_613_800,
             "official_name": "VideoClassifier",
             "path": "video_classifier",
         },
     },
     "videoswin_base_kinetics_classifier": {
         "metadata": {
-            "description": ("videoswin_base_kinetics "),  # TODO: update
-            "params": 25_613_800,  # TODO: update
+            "description": ("videoswin_base_kinetics "),
+            "params": 25_613_800,
             "official_name": "VideoClassifier",
             "path": "video_classifier",
         },
     },
     "videoswin_base_something_something_v2_classifier": {
         "metadata": {
             "description": (
-                "videoswin_base_something_something_v2 "  # TODO: update
+                "videoswin_base_something_something_v2 "
             ),
-            "params": 25_613_800,  # TODO: update
+            "params": 25_613_800,
             "official_name": "VideoClassifier",
             "path": "video_classifier",
         },