Explicitly specify full package for more modules, as shown in #2001 (#2077)

* Test to demonstrate failures in import

* Explicitly specify full package for more modules, as shown in #2001

Author: BjarneHerland

keras_cv/datasets/imagenet/load.py

@@ -56,7 +56,9 @@ def apply(example):
     return apply
 
 
-@keras_cv_export("keras_cv.datasets.imagenet.load")
+@keras_cv_export(
+    "keras_cv.datasets.imagenet.load", package="keras_cv.datasets.imagenet"
+)
 def load(
     split,
     tfrecord_path,

keras_cv/datasets/load_test.py

@@ -0,0 +1,24 @@
+# Copyright 2023 The KerasCV Authors
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     https://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""Tests ability to load keras_cv.datasets."""
+
+from keras_cv.tests.test_case import TestCase
+
+
+class LoadTest(TestCase):
+    def test_datasets_load(self):
+        import keras_cv.datasets  # noqa: F401
+
+    def test_imagenet_load(self):
+        import keras_cv.datasets.imagenet  # noqa: F401

keras_cv/datasets/pascal_voc/segmentation.py

@@ -465,7 +465,10 @@ def md_gen():
     return dataset
 
 
-@keras_cv_export("keras_cv.datasets.pascal_voc.segmentation.load")
+@keras_cv_export(
+    "keras_cv.datasets.pascal_voc.segmentation.load",
+    package="keras_cv.datasets.pascal_voc_segmentation",
+)
 def load(
     split="sbd_train",
     data_dir=None,

keras_cv/keypoint/converters.py

@@ -60,7 +60,9 @@ def _xy_noop(keypoints, images=None):
 }
 
 
-@keras_cv_export("keras_cv.keypoint.convert_format")
+@keras_cv_export(
+    "keras_cv.keypoint.convert_format", package="keras_cv.keypoint"
+)
 def convert_format(keypoints, source, target, images=None, dtype=None):
     """Converts keypoints from one format to another.