Fix python lints for generate_test_torchscripts.py (#25107)
Summary:
Fix lints, checked with flake8
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25107
Reviewed By: zrphercule
Differential Revision: D16991296
Pulled By: IvanKobzarev
fbshipit-source-id: 5b69d716e3c458dc2cfe5b668a390c7272b1c74f
diff --git a/android/pytorch_android/generate_test_torchscripts.py b/android/pytorch_android/generate_test_torchscripts.py
index d95d668..b393d7c 100644
--- a/android/pytorch_android/generate_test_torchscripts.py
+++ b/android/pytorch_android/generate_test_torchscripts.py
@@ -1,16 +1,15 @@
import torch
-from typing import List, Tuple, Dict
OUTPUT_DIR = "src/androidTest/assets/"
def scriptAndSave(module, fileName):
- print('-'*80)
+ print('-' * 80)
script_module = torch.jit.script(module)
print(script_module.graph)
outputFileName = OUTPUT_DIR + fileName
script_module.save(outputFileName)
print("Saved to " + outputFileName)
- print('='*80)
+ print('=' * 80)
class Test(torch.jit.ScriptModule):
def __init__(self):