Trying a new way to generate way to generate files.

1 files changed, 7 insertions, 2 deletions

diff --git a/src/ee/__main__.py b/src/ee/__main__.py
index 6a7517f..a387b1d 100644
--- a/src/ee/__main__.py
+++ b/src/ee/__main__.py
@@ -1,9 +1,11 @@
-from functools import total_ordering
-import ee.tools
 import importlib
 import logging
 import pkgutil
 import sys
+import os
+from functools import total_ordering
+
+import ee.tools
 
 
 def eprint(*args, **kwargs):
@@ -50,6 +52,9 @@ def main():
     name = sys.argv[1]
     del sys.argv[1]
 
+    if os.path.isdir("py"):
+        sys.path.append("py")
+
     for t in tools:
         if t.name != name:
             continue