fix: template file extension rename to .py (#43) [skip ci]

* fix: file ext to pyi for image classification

* fix: rename to .py

Author: Jeff Yang

.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

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
 

templates/_base/_argparse.pyi renamed to templates/_base/_argparse.py

@@ -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 }}

templates/_base/_events.pyi renamed to templates/_base/_events.py

@@ -38,4 +38,4 @@ def dependencies(fname):
     ),
     zip_safe=True,
     install_requires=requirements,
-)
+)