From a582ceccaad630401251b64a24a679e1fabbfcb0 Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Mon, 29 Apr 2024 15:24:57 -0700 Subject: [PATCH 1/2] Export APIs --- .../src/models/backbones/video_swin/video_swin_aliases.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py index 781b718910..42b4ce46eb 100644 --- a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py +++ b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py @@ -14,6 +14,7 @@ import copy +from keras_cv.api_export import keras_cv_export from keras_cv.src.models.backbones.video_swin.video_swin_backbone import ( VideoSwinBackbone, ) @@ -42,6 +43,7 @@ """ # noqa: E501 +@keras_cv_export("keras_cv.models.VideoSwinTBackbone") class VideoSwinTBackbone(VideoSwinBackbone): def __new__( cls, @@ -79,6 +81,7 @@ def presets_with_weights(cls): return cls.presets +@keras_cv_export("keras_cv.models.VideoSwinSBackbone") class VideoSwinSBackbone(VideoSwinBackbone): def __new__( cls, @@ -116,6 +119,7 @@ def presets_with_weights(cls): return cls.presets +@keras_cv_export("keras_cv.models.VideoSwinBBackbone") class VideoSwinBBackbone(VideoSwinBackbone): def __new__( cls, From 41b7b5ee4be942adbe16bcd53b4d5b0c3576576e Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Mon, 29 Apr 2024 22:37:38 +0000 Subject: [PATCH 2/2] generate API --- keras_cv/api/models/__init__.py | 9 +++++++++ .../models/backbones/video_swin/video_swin_aliases.py | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/keras_cv/api/models/__init__.py b/keras_cv/api/models/__init__.py index e425b26899..97f9bc577b 100644 --- a/keras_cv/api/models/__init__.py +++ b/keras_cv/api/models/__init__.py @@ -178,6 +178,15 @@ ResNetV2Backbone, ) from keras_cv.src.models.backbones.vgg16.vgg16_backbone import VGG16Backbone +from keras_cv.src.models.backbones.video_swin.video_swin_aliases import ( + VideoSwinBBackbone, +) +from keras_cv.src.models.backbones.video_swin.video_swin_aliases import ( + VideoSwinSBackbone, +) +from keras_cv.src.models.backbones.video_swin.video_swin_aliases import ( + VideoSwinTBackbone, +) from keras_cv.src.models.backbones.video_swin.video_swin_backbone import ( VideoSwinBackbone, ) diff --git a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py index 42b4ce46eb..3bbc4a4f85 100644 --- a/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py +++ b/keras_cv/src/models/backbones/video_swin/video_swin_aliases.py @@ -14,7 +14,7 @@ import copy -from keras_cv.api_export import keras_cv_export +from keras_cv.src.api_export import keras_cv_export from keras_cv.src.models.backbones.video_swin.video_swin_backbone import ( VideoSwinBackbone, )