Simplify loading lua classes: add `PyTorchHelpers.load_lua_class(lua_…

…filename, lua_classname)`

setup.py

@@ -139,7 +139,7 @@ def get_file_datetime(filepath):
 
 setup(
     name='PyTorch',
-    version='2.5.0',
+    version='2.6.0',
     author='Hugh Perkins',
     author_email='[email protected]',
     description=(
@@ -152,7 +152,7 @@ def get_file_datetime(filepath):
     install_requires=['numpy'],
     scripts=[],
     ext_modules=ext_modules,
-    py_modules=['floattensor', 'PyTorchAug'],
+    py_modules=['floattensor', 'PyTorchAug', 'PyTorchHelpers'],
     package_dir={'': 'src'}
 )
 

simpleexample/pybit.py

@@ -1,18 +1,10 @@
 import sys
 import os
-import PyTorchAug
 import PyTorch
+import PyTorchHelpers
 import numpy as np
 
-PyTorch.require('luabit')
-class Luabit(PyTorchAug.LuaClass):
-    def __init__(self, _fromLua=False):
-        self.luaclass = 'Luabit'
-        if not _fromLua:
-            name = self.__class__.__name__
-            super(self.__class__, self).__init__([name])
-        else:
-            self.__dict__['__objectId'] = getNextObjectId()
+Luabit = PyTorchHelpers.load_lua_class('luabit.lua', 'Luabit')
 
 batchSize = 2
 numFrames = 4

src/PyTorchHelpers.py

@@ -0,0 +1,16 @@
+import PyTorch
+import PyTorchAug
+
+def load_lua_class(lua_filename, lua_classname):
+    module = lua_filename.replace('.lua', '')
+    PyTorch.require(module)
+    class LuaWrapper(PyTorchAug.LuaClass):
+        def __init__(self, _fromLua=False):
+            self.luaclass = lua_classname
+            if not _fromLua:
+                name = lua_classname
+                super(self.__class__, self).__init__([name])
+            else:
+                self.__dict__['__objectId'] = getNextObjectId()
+    return LuaWrapper
+