fix package name

lucidrains · Jun 4, 2024 · 183a860 · 183a860
1 parent 35cd387
commit 183a860
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/frame_averaging_pytorch/frame_averaging.py b/frame_averaging_pytorch/frame_averaging.py
@@ -193,4 +193,3 @@ def frame_average(out):
  return out
 
  return (out, *rest)
-
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "frame-averaging-pytorch"
-version = "0.0.16"
+version = "0.0.17"
 description = "Frame Averaging"
 authors = [
  { name = "Phil Wang", email = "[email protected]" }
@@ -25,7 +25,7 @@ classifiers=[
 dependencies = [
  "torch>=2.0",
  "einops>=0.8.0",
- 'optree',
+ "optree"
 ]
 
 [project.urls]
@@ -47,4 +47,4 @@ dev-dependencies = []
 allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
-packages = ["frame-averaging-pytorch"]
+packages = ["frame_averaging_pytorch"]