fixed setup changes

wildoctopus · May 12, 2023 · d5ba7b0 · d5ba7b0
1 parent 9d5d4ba
commit d5ba7b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -18,7 +18,7 @@ def get_requirements():
 
 def get_version():
     current_dir = os.path.abspath(os.path.dirname(__file__))
-    version_file = os.path.join(current_dir, "cb_loss", "__init__.py")
+    version_file = os.path.join(current_dir, "cbloss", "__init__.py")
     with io.open(version_file, encoding="utf-8") as f:
         return re.search(r'^__version__ = [\'"]([^\'"]*)[\'"]', f.read(), re.M).group(1)