Pārlūkot izejas kodu

Update src/client/remote_sequential.py

justheuristic 3 gadi atpakaļ
vecāks
revīzija
052116adb6
1 mainītis faili ar 3 papildinājumiem un 0 dzēšanām
  1. 3 0
      src/client/remote_sequential.py

+ 3 - 0
src/client/remote_sequential.py

@@ -94,3 +94,6 @@ class RemoteTransformerBlock(RemoteSequential):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         assert len(self) == 1, "Remote Block is a sequence size 1"
+        
+    def extra_repr(self):
+        return f"{self.sequence_manager.block_uids[0]}"