Update pytorch.js

lutzroeder · Oct 12, 2024 · b170b44 · b170b44
1 parent 0cb3c3c
commit b170b44
Show file tree

Hide file tree

Showing 5 changed files with 1,627 additions and 528 deletions.
diff --git a/source/python.js b/source/python.js
@@ -6213,16 +6213,29 @@ python.Execution = class {
             }
         });
         this.registerType('torch.TupleType', class extends torch.Type {
-            constructor() {
+            constructor(elements) {
                 super('TupleType');
+                this._elements = elements;
+            }
+            elements() {
+                return this._elements;
             }
         });
         this.registerType('torch.TensorType', class extends torch.Type {
             constructor() {
                 super('TensorType');
             }
         });
-        this.registerType('torch.AnyType', class extends torch.Type {});
+        this.registerType('torch.AnyType', class extends torch.Type {
+            constructor() {
+                super('AnyType');
+            }
+        });
+        this.registerType('torch.NoneType', class extends torch.Type {
+            constructor() {
+                super('NoneType');
+            }
+        });
         this.registerType('torch.NumberType', class extends torch.Type {
             constructor() {
                 super('NumberType');