Setup geometry_script module if the module name is different

__init__.py

@@ -16,6 +16,10 @@ import os
 
 from .tree import *
 
+# Set the `geometry_script` module to the current module in case the folder is named differently.
+import sys
+sys.modules['geometry_script'] = sys.modules[__name__]
+
 bl_info = {
     "name" : "Geometry Script",
     "author" : "Carson Katri",

examples/repeat.py

@@ -1,11 +0,0 @@
-from geometry_script import *
-
-@tree
-def repeat(size: Vector):
-    cubes = []
-    for x in range(5):
-        for y in range(5):
-            for z in range(5):
-                c = cube(size=(0.1, 0.1, 0.1))
-                cubes.append(c.transform(translation=(x, y, z)))
-    return join_geometry(geometry=cubes)