Add back info in setup.py to upload to PyPI

mamba_ssm/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.2.3"
+__version__ = "2.2.3.post1"
 
 from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
 from mamba_ssm.modules.mamba_simple import Mamba

setup.py

@@ -348,13 +348,31 @@ def run(self):
             "mamba_ssm.egg-info",
         )
     ),
+    author="Tri Dao, Albert Gu",
+    author_email="[email protected], [email protected]",
+    description="Mamba state-space model",
     long_description=long_description,
     long_description_content_type="text/markdown",
-
+    url="https://github.com/state-spaces/mamba",
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: BSD License",
+        "Operating System :: Unix",
+    ],
     ext_modules=ext_modules,
     cmdclass={"bdist_wheel": CachedWheelsCommand, "build_ext": BuildExtension}
     if ext_modules
     else {
         "bdist_wheel": CachedWheelsCommand,
-    }
+    },
+    python_requires=">=3.9",
+    install_requires=[
+        "torch",
+        "packaging",
+        "ninja",
+        "einops",
+        "triton",
+        "transformers",
+        # "causal_conv1d>=1.4.0",
+    ],
 )