bump up version and gate magic-wand version

setup.py

@@ -411,12 +411,6 @@ def _read_requirements(filename: str) -> List[str]:
 if not _is_neuron():
     ext_modules.append(CMakeExtension(name="vllm._C"))
 
-# UPSTREAM SYNC: needed for sparsity
-_sparsity_deps = ["nm-magic-wand-nightly"]
-nm_release_type = os.getenv(NM_RELEASE_TYPE)
-if nm_release_type == 'RELEASE':
-    _sparsity_deps = ["nm-magic-wand"]
-
 package_data = {
     "vllm": ["py.typed", "model_executor/layers/fused_moe/configs/*.json"]
 }
@@ -458,6 +452,11 @@ def _read_requirements(filename: str) -> List[str]:
     python_requires=">=3.8",
     install_requires=get_requirements(),
     ext_modules=ext_modules,
+    # UPSTREAM SYNC: needed for sparsity
+    _sparsity_deps = [f"nm-magic-wand-nightly~={version}"]
+    nm_release_type = os.getenv(NM_RELEASE_TYPE)
+    if nm_release_type == 'RELEASE':
+        _sparsity_deps = [f"nm-magic-wand~={version}"]
     extras_require={
         "tensorizer": ["tensorizer==2.9.0"],
         # UPSTREAM SYNC: required for sparsity

vllm/__init__.py

@@ -10,7 +10,7 @@
 from vllm.sampling_params import SamplingParams
 
 # UPSTREAM SYNC: use the current downstream.
-__version__ = "0.3.0"
+__version__ = "0.4.0"
 
 __all__ = [
     "LLM",