From 9d9fd57fe68014cf6e80f3cdd33c02f00ffbb0ba Mon Sep 17 00:00:00 2001
From: Carson Katri <carson.katri@gmail.com>
Date: Sun, 13 Nov 2022 15:58:25 -0500
Subject: [PATCH] Setup geometry_script module if the module name is different

---
 __init__.py                          |  4 ++++
 examples/{Cube.py => Repeat Grid.py} |  0
 examples/repeat.py                   | 11 -----------
 3 files changed, 4 insertions(+), 11 deletions(-)
 rename examples/{Cube.py => Repeat Grid.py} (100%)
 delete mode 100644 examples/repeat.py

diff --git a/__init__.py b/__init__.py
index 35309a7..a1d140c 100644
--- a/__init__.py
+++ b/__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",
diff --git a/examples/Cube.py b/examples/Repeat Grid.py
similarity index 100%
rename from examples/Cube.py
rename to examples/Repeat Grid.py
diff --git a/examples/repeat.py b/examples/repeat.py
deleted file mode 100644
index f6df584..0000000
--- a/examples/repeat.py
+++ /dev/null
@@ -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)
\ No newline at end of file