Skip to content

Commit 8067bd6

Browse files
committed
fix format issue
1 parent 38ece90 commit 8067bd6

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

test/quantization/test_quant_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -499,6 +499,7 @@ def test_quantized_tensor_subclass_save_load_map_location(self):
499499
@unittest.skipIf(not torch.accelerator.is_available(), "Need GPU available")
500500
def test_quantized_model_streaming(self):
501501
device_module = torch.get_device_module(_DEVICE)
502+
502503
def reset_memory():
503504
gc.collect()
504505
device_module.empty_cache()
@@ -1110,6 +1111,7 @@ def test_non_fqn_config_filter_fn_none(self):
11101111
@unittest.skipIf(not torch.accelerator.is_available(), "Need GPU available")
11111112
def test_quantized_model_streaming_fqn_config(self):
11121113
device_module = torch.get_device_module(_DEVICE)
1114+
11131115
def reset_memory():
11141116
gc.collect()
11151117
device_module.empty_cache()

test/quantization/test_quant_primitives.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@
4343

4444
_DEVICE = auto_detect_device()
4545

46+
4647
# Helper function to run a function twice
4748
# and verify that the result is the same.
4849
# Adds some verification to avoid side effects.

0 commit comments

Comments
 (0)