frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-04 08:00:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE] Enable more flake8-comprehensions checks (#94601)

Browse Source 
I applied some flake8 fixes and enabled checking for them in the linter. I also enabled some checks for my previous comprehensions PR.

This is a follow up to #94323 where I enable the flake8 checkers for the fixes I made and fix a few more of them.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/94601
Approved by: https://github.com/ezyang
This commit is contained in:
Aaron Gokaslan
2023-02-10 23:40:26 +00:00
committed by PyTorch MergeBot
parent 0b31ebf9e4
commit 3d82d8d0ed
30 changed files with 71 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/quantization/fx/test_model_report_fx.py
View File

@ -900,7 +900,7 @@ class TestFxModelReportClass(QuantizationTestCase):
model_report = ModelReport(model_prep, test_detector_set)
# make sure internal valid reports matches
detector_name_set = set([detector.get_detector_name() for detector in test_detector_set])
detector_name_set = {detector.get_detector_name() for detector in test_detector_set}
self.assertEqual(model_report.get_desired_reports_names(), detector_name_set)
# now attempt with no valid reports, should raise error
@ -1329,7 +1329,7 @@ class TestFxDetectInputWeightEqualization(QuantizationTestCase):
mods_to_check = set([nn.Linear, nn.Conv2d])
# get the set of all nodes in the graph their fqns
node_fqns = set([node.target for node in prepared_for_callibrate_model.graph.nodes])
node_fqns = {node.target for node in prepared_for_callibrate_model.graph.nodes}
# there should be 4 node fqns that have the observer inserted
correct_number_of_obs_inserted = 4