Skip to content

Commit

Permalink
add back keras_cv
Browse files Browse the repository at this point in the history
  • Loading branch information
divyashreepathihalli committed Sep 27, 2024
1 parent 6a339bc commit 89da3b2
Show file tree
Hide file tree
Showing 12 changed files with 887 additions and 0 deletions.
1 change: 1 addition & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@ jupyter
pydot
boto3
tensorflow
keras-cv
keras-tuner
tf_keras
keras-hub-nightly # TODO: update to keras-hub.
2 changes: 2 additions & 0 deletions scripts/api_master.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from kt_api_master import KT_API_MASTER
from cv_api_master import CV_API_MASTER
from hub_api_master import HUB_API_MASTER

API_MASTER = {
Expand Down Expand Up @@ -1821,6 +1822,7 @@
],
},
KT_API_MASTER,
CV_API_MASTER,
HUB_API_MASTER,
],
}
2 changes: 2 additions & 0 deletions scripts/autogen.py
Original file line number Diff line number Diff line change
Expand Up @@ -539,6 +539,8 @@ def make_md_source_for_entry(self, entry, path_stack, title_stack):
"missing {{toc}} tag." % (template_path,)
)
template = template.replace("{{toc}}", toc)
if "keras_cv/" in path_stack and "models/" in path_stack:
template = render_tags.render_tags(template, keras_cv)
if "keras_hub/" in path_stack and "models/" in path_stack:
template = render_tags.render_tags(template, keras_hub)
source_path = Path(self.md_sources_dir) / Path(*path_stack)
Expand Down
Loading

0 comments on commit 89da3b2

Please sign in to comment.