improve: cookiecutter working

Author: samedii

pyproject.toml

@@ -4,6 +4,7 @@ version = "0.0.0"
 description = "Pytorch project template and related tools"
 authors = ["Aiwizo"]
 license = "Apache 2"
+include = ["workflow/template/*"]
 
 [tool.poetry.dependencies]
 python = "3.8"

test_template.sh

@@ -1,7 +1,12 @@
 #!/usr/bin/env bash
 set -o errexit -o nounset -o pipefail -o xtrace
 
+rm dist/workflow-0.0.0-py3-none-any.whl
+poetry build
 cd test_template
+poetry remove workflow
+poetry install --remove-untracked
+poetry add ../dist/workflow-0.0.0-py3-none-any.whl
 shopt -s extglob 
 rm -rf .github
 rm -rf !(pyproject.toml)

workflow/__init__.py

@@ -1,10 +1,10 @@
+from workflow import functional
+
 from workflow.figure_to_numpy import figure_to_numpy
 from workflow.numpy_seed import numpy_seed
 from workflow.progress_bar import ProgressBar
 from workflow.early_stopping import EarlyStopping
 
-from workflow import functional
-from workflow import ignite
 from workflow import torch
 
 from pkg_resources import get_distribution, DistributionNotFound

workflow/functional/__init__.py

@@ -1,5 +1,3 @@
-from .interleaved import interleaved
-from .repeat_map_chain import repeat_map_chain
-from .star import star
-from .starcompose import starcompose
-from .structure_map import structure_map
+from workflow.functional.star import star
+from workflow.functional.starcompose import starcompose
+from workflow.functional.structure_map import structure_map

workflow/setup_project.py

@@ -6,15 +6,14 @@
 
 
 if __name__ == '__main__':
+    print(__file__ + '/template')
     try:
         repository_path = cookiecutter(
-            '/'.join(__file__.split('/')[:-5]) + '/workflow/template',
-        )
-    except RepositoryNotFound:
-        # Probably installed with the -e flag
-        repository_path = cookiecutter(
-            '/'.join(__file__.split('/')[:-2]) + '/template',
+            str(Path(__file__).parent / 'template')
         )
+    except RepositoryNotFound as exception:
+        print('__file__:', __file__)
+        raise exception
 
     repository_path = Path(repository_path)
 

template/cookiecutter.json renamed to workflow/template/cookiecutter.json

@@ -1,7 +1,3 @@
-from workflow.torch.is_float import is_float
-from workflow.torch.get_conv_output_size import get_conv_output_size
-from workflow.torch.get_model_summary import get_model_summary
-from workflow.torch.initialize_weights import initialize_weights
 from workflow.torch.module_device import module_device
 from workflow.torch.module_compose import ModuleCompose
 from workflow.torch.to_device import to_device