Skip to content

Commit 4031ac6

Browse files
committed
fix: add tests
1 parent aa82461 commit 4031ac6

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

tests/tests_fabric/test_connector.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -178,6 +178,10 @@ def parse_devices(devices):
178178
@staticmethod
179179
def get_parallel_devices(devices):
180180
return [torch.device("cpu")] * devices
181+
182+
@staticmethod
183+
def get_device() -> str:
184+
return "cpu"
181185

182186
@staticmethod
183187
def auto_device_count() -> int:

tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,6 +191,10 @@ def parse_devices(devices):
191191
@staticmethod
192192
def get_parallel_devices(devices):
193193
return [torch.device("cpu")] * devices
194+
195+
@staticmethod
196+
def get_device() -> str:
197+
return "cpu"
194198

195199
@staticmethod
196200
def auto_device_count() -> int:

0 commit comments

Comments
 (0)