Skip to content

Commit d4447a4

Browse files
fix(envoy): correct property decorator for devices method
- Added the missing @Property decorator to the devices method in the Envoy class, ensuring proper functionality and access to device information.
1 parent 22901c0 commit d4447a4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/nnsight/intervention/envoy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -673,7 +673,7 @@ def device(self) -> Optional[torch.device]:
673673
except:
674674
return None
675675

676-
property
676+
@property
677677
def devices(self) -> Optional[set[torch.device]]:
678678
"""
679679
Get the devices the module is on. Finds all parameters and return their devices.

0 commit comments

Comments
 (0)