Necessary fixes to get it to work

codes/trainer/inject.py

@@ -3,6 +3,7 @@ import inspect
 import pkgutil
 import re
 import sys
+import os
 
 import torch.nn
 
@@ -33,7 +34,8 @@ def format_injector_name(name):
 # Works by loading all python modules in the injectors/ directory and sniffing out subclasses of Injector.
 # field will be properly populated.
 def find_registered_injectors(base_path="trainer/injectors"):
-    module_iter = pkgutil.walk_packages([base_path])
+    path = os.path.normpath(os.path.join(os.path.dirname(os.path.realpath(__file__)), f'../{base_path}'))
+    module_iter = pkgutil.walk_packages([path])
     results = {}
     for mod in module_iter:
         if mod.ispkg:

codes/trainer/networks.py

@@ -30,14 +30,12 @@ def register_model(func):
     func._dlas_registered_model = True
     return func
 
-
 def find_registered_model_fns(base_path='models'):
     found_fns = {}
-    module_iter = pkgutil.walk_packages([base_path])
+    path = os.path.normpath(os.path.join(os.path.dirname(os.path.realpath(__file__)), f'../{base_path}'))
+
+    module_iter = pkgutil.walk_packages([path])
     for mod in module_iter:
-        if os.name == 'nt':
-            if os.path.join(os.getcwd(), base_path) not in mod.module_finder.path:
-                continue   # I have no idea why this is necessary - I think it's a bug in the latest PyWindows release.
         if mod.ispkg:
             EXCLUSION_LIST = ['flownet2']
             if mod.name not in EXCLUSION_LIST:

setup.py

@@ -1,19 +1,13 @@
 import setuptools
 
+from pip.req import parse_requirements
+
 with open("README.old.md", "r", encoding="utf-8") as fh:
     long_description = fh.read()
 
-# kludge
-packages = setuptools.find_packages()
-for i in range(len(packages)):
-    packages[i] = packages[i].replace("codes", "dlas")
-
 setuptools.setup(
     name="DL-Art-School",
-    packages=packages,
-    package_dir={
-        "dlas": "./codes"
-    },
+    packages=setuptools.find_packages(),
     version="0.0.1",
     author="James Betker",
     author_email="james@adamant.ai",
@@ -24,7 +18,7 @@ setuptools.setup(
     project_urls={},
     scripts=[],
     include_package_data=True,
-    install_requires=[],
+    install_requires=parse_requirements('requirements.txt', session='hack'),
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: Apache Software License",