pytorch-ignite · ydcjeff · Apr 6, 2021 · Apr 6, 2021 · Apr 6, 2021
diff --git a/.github/run_code_style.sh b/.github/run_code_style.sh
@@ -3,12 +3,12 @@
 set -xeu
 
 if [ $1 == "lint" ]; then
-    flake8 app templates tests --config .flake8
-    isort app templates tests --check --settings pyproject.toml
-    black app templates tests --check --config pyproject.toml
+    flake8 app templates/*/_sidebar.py tests --config .flake8
+    isort app templates/*/_sidebar.py tests --check --settings pyproject.toml
+    black app templates/*/_sidebar.py tests --check --config pyproject.toml
 elif [ $1 == "fmt" ]; then
-    isort app templates tests --color --settings pyproject.toml
-    black app templates tests --config pyproject.toml
+    isort app templates/*/_sidebar.py tests --color --settings pyproject.toml
+    black app templates/*/_sidebar.py tests --config pyproject.toml
 elif [ $1 == "install" ]; then
     pip install flake8 "black==20.8b1" "isort==5.7.0"
 fi
diff --git a/app/codegen.py b/app/codegen.py
@@ -25,7 +25,7 @@ def render_templates(self, template_name: str, project_name: str, config: dict):
         )
         for fname in env.list_templates(filter_func=lambda x: not x.startswith("_")):
             code = env.get_template(fname).render(**config)
-            fname = fname.replace(".pyi", ".py").replace(template_name, project_name)
+            fname = fname.replace(template_name, project_name)
             self.rendered_code[fname] = code
             yield fname, code
 

diff --git a/templates/_base/_argparse.pyi → templates/_base/_argparse.py b/templates/_base/_argparse.pyi → templates/_base/_argparse.py
diff --git a/templates/_base/_events.pyi → templates/_base/_events.py b/templates/_base/_events.pyi → templates/_base/_events.py
diff --git a/templates/_base/_handlers.pyi → templates/_base/_handlers.py b/templates/_base/_handlers.pyi → templates/_base/_handlers.py
diff --git a/templates/_base/_hubconf.pyi → templates/_base/_hubconf.py b/templates/_base/_hubconf.pyi → templates/_base/_hubconf.py
diff --git a/templates/gan/gan/__init__.pyi → templates/gan/gan/__init__.py b/templates/gan/gan/__init__.pyi → templates/gan/gan/__init__.py
diff --git a/templates/gan/gan/config.pyi → templates/gan/gan/config.py b/templates/gan/gan/config.pyi → templates/gan/gan/config.py
diff --git a/templates/gan/gan/datasets.pyi → templates/gan/gan/datasets.py b/templates/gan/gan/datasets.pyi → templates/gan/gan/datasets.py
diff --git a/templates/gan/gan/handlers.pyi → templates/gan/gan/handlers.py b/templates/gan/gan/handlers.pyi → templates/gan/gan/handlers.py
diff --git a/templates/gan/gan/main.pyi → templates/gan/gan/main.py b/templates/gan/gan/main.pyi → templates/gan/gan/main.py
diff --git a/templates/gan/gan/models.pyi → templates/gan/gan/models.py b/templates/gan/gan/models.pyi → templates/gan/gan/models.py
diff --git a/templates/gan/gan/trainers.pyi → templates/gan/gan/trainers.py b/templates/gan/gan/trainers.pyi → templates/gan/gan/trainers.py
diff --git a/templates/gan/gan/utils.pyi → templates/gan/gan/utils.py b/templates/gan/gan/utils.pyi → templates/gan/gan/utils.py
diff --git a/templates/gan/hubconf.pyi → templates/gan/hubconf.py b/templates/gan/hubconf.pyi → templates/gan/hubconf.py
diff --git a/templates/gan/requirements.txt b/templates/gan/requirements.txt
@@ -3,6 +3,6 @@ torch>=1.7.0
 pytorch-ignite>=0.4.4
 torchvision>=0.8.0
 matplotlib>=3.4.1
-pandas>=1.2.3
+pandas
 {{ handler_deps }}
 {{ logger_deps }}
diff --git a/templates/gan/tests/test_datasets.pyi → templates/gan/tests/test_datasets.py b/templates/gan/tests/test_datasets.pyi → templates/gan/tests/test_datasets.py
diff --git a/templates/gan/tests/test_handlers.pyi → templates/gan/tests/test_handlers.py b/templates/gan/tests/test_handlers.pyi → templates/gan/tests/test_handlers.py
diff --git a/templates/gan/tests/test_models.pyi → templates/gan/tests/test_models.py b/templates/gan/tests/test_models.pyi → templates/gan/tests/test_models.py
diff --git a/templates/gan/tests/test_trainers.pyi → templates/gan/tests/test_trainers.py b/templates/gan/tests/test_trainers.pyi → templates/gan/tests/test_trainers.py
diff --git a/templates/gan/tests/test_utils.pyi → templates/gan/tests/test_utils.py b/templates/gan/tests/test_utils.pyi → templates/gan/tests/test_utils.py
diff --git a/templates/image_classification/setup.py b/templates/image_classification/setup.py
@@ -38,4 +38,4 @@ def dependencies(fname):
     ),
     zip_safe=True,
     install_requires=requirements,
-)
+)
diff --git a/templates/single/hubconf.pyi → templates/single/hubconf.py b/templates/single/hubconf.pyi → templates/single/hubconf.py
diff --git a/templates/single/single/__init__.pyi → templates/single/single/__init__.py b/templates/single/single/__init__.pyi → templates/single/single/__init__.py
diff --git a/templates/single/single/config.pyi → templates/single/single/config.py b/templates/single/single/config.pyi → templates/single/single/config.py
diff --git a/templates/single/single/datasets.pyi → templates/single/single/datasets.py b/templates/single/single/datasets.pyi → templates/single/single/datasets.py
diff --git a/templates/single/single/handlers.pyi → templates/single/single/handlers.py b/templates/single/single/handlers.pyi → templates/single/single/handlers.py
diff --git a/templates/single/single/main.pyi → templates/single/single/main.py b/templates/single/single/main.pyi → templates/single/single/main.py
diff --git a/templates/single/single/models.pyi → templates/single/single/models.py b/templates/single/single/models.pyi → templates/single/single/models.py
diff --git a/templates/single/single/trainers.pyi → templates/single/single/trainers.py b/templates/single/single/trainers.pyi → templates/single/single/trainers.py
diff --git a/templates/single/single/utils.pyi → templates/single/single/utils.py b/templates/single/single/utils.pyi → templates/single/single/utils.py
diff --git a/templates/single/tests/test_handlers.pyi → templates/single/tests/test_handlers.py b/templates/single/tests/test_handlers.pyi → templates/single/tests/test_handlers.py
diff --git a/templates/single/tests/test_trainers.pyi → templates/single/tests/test_trainers.py b/templates/single/tests/test_trainers.pyi → templates/single/tests/test_trainers.py
diff --git a/templates/single/tests/test_utils.pyi → templates/single/tests/test_utils.py b/templates/single/tests/test_utils.pyi → templates/single/tests/test_utils.py
-Original file line number
+Diff line change
@@ Expand Up / @@ -38,4 +38,4 @@ def dependencies(fname): @@
         ),
         zip_safe=True,
         install_requires=requirements,
-    )
+    )