Use black for lib/models

2021-05-12 16:28:05 +08:00
parent d51e5fdc7f
commit f1c47af5fa
42 changed files with 7552 additions and 4688 deletions
--- a/lib/models/cell_searchs/_test_module.py
+++ b/lib/models/cell_searchs/_test_module.py
@@ -4,9 +4,11 @@
 import torch
 from search_model_enas_utils import Controller

-def main():
-  controller = Controller(6, 4)
-  predictions = controller()

-if __name__ == '__main__':
-  main()
+def main():
+    controller = Controller(6, 4)
+    predictions = controller()
+
+
+if __name__ == "__main__":
+    main()