bug fix for version update

plemeri · Sep 26, 2023 · 359fb1b · 359fb1b
1 parent b5230c7
commit 359fb1b
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/MANIFEST.in b/MANIFEST.in
@@ -0,0 +1 @@
+include transparent_background/config.yaml
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="transparent-background",
-    version="1.2.5",
+    version="1.2.7",
     author="Taehun Kim",
     author_email="taehoon1018@postech.ac.kr",
     description="Make images with transparent background",
@@ -27,5 +27,5 @@
             "transparent-background=transparent_background:console",
         ],
     },
-    package_data={'transparent-background': ['~', '.transparent-background', 'config.yaml']}
+    include_package_data=True
 )
diff --git a/transparent_background/Remover.py b/transparent_background/Remover.py
@@ -39,7 +39,7 @@ def __init__(self, mode="base", jit=False, device=None, ckpt=None, fast=None):
         os.makedirs(home_dir, exist_ok=True)
 
         if not os.path.isfile(os.path.join(home_dir, "config.yaml")):
-            shutil.copy(os.path.join(os.path.curdir, "config.yaml"), os.path.join(home_dir, "config.yaml"))
+            shutil.copy(os.path.join(repopath, "config.yaml"), os.path.join(home_dir, "config.yaml"))
         self.meta = load_config(os.path.join(home_dir, "config.yaml"))[mode]
 
         if fast is not None:

diff --git a/config.yaml → transparent_background/config.yaml b/config.yaml → transparent_background/config.yaml