Update pytorch.js

lutzroeder · Oct 11, 2024 · db7ca37 · db7ca37
1 parent b21850d
commit db7ca37
Show file tree

Hide file tree

Showing 5 changed files with 2,648 additions and 592 deletions.
diff --git a/source/python.js b/source/python.js
@@ -6163,7 +6163,7 @@ python.Execution = class {
         });
         this.registerType('torch.ClassType', class extends torch.Type {
             constructor(qualified_name, cu, is_module) {
-                super();
+                super('ClassType');
                 this._qualified_name = qualified_name;
                 this._is_module = is_module;
                 this._attributes = new Map();
@@ -6328,6 +6328,9 @@ python.Execution = class {
             getValueType() {
                 return this._value;
             }
+            __str__() {
+                return `Dict(${this.getKeyType().toString()}, ${this.getValueType().toString()})`;
+            }
         });
         this.registerType('torch.DeviceObjType', class extends torch.Type {
             constructor() {