FIX Docs minor refactoring for successful build.

petrasvestartas · petrasvestartas · commit 246cd9a3ad1a · 2025-01-06T11:28:09.000+01:00
diff --git a/docs/api/compas_model.notebook.rst b/docs/api/compas_model.notebook.rst
@@ -12,5 +12,4 @@ Classes
     :nosignatures:
 
     ThreeBlockObject
-    ThreeModelObject
-    ThreeElementObject
+    ThreeModelObject
diff --git a/src/compas_model/notebook/__init__.py b/src/compas_model/notebook/__init__.py
@@ -0,0 +1,6 @@
+from .scene import ThreeBlockObject, ThreeModelObject
+
+__all__ = [
+    "ThreeBlockObject",
+    "ThreeModelObject",
+]
diff --git a/src/compas_model/notebook/scene/__init__.py b/src/compas_model/notebook/scene/__init__.py
@@ -20,10 +20,3 @@ def register_scene_objects():
     register(Model, ThreeModelObject, context="Notebook")
 
     # print("PyThreeJS Model elements registered.")
-
-
-__all__ = [
-    "ThreeBlockObject",
-    "ThreeModelObject",
-    "ThreeElementObject",
-]
