fix: gpu memory service cli not found (#5749)

2ef408ff · Schwinn Saereesitthipitak · GitHub · e2159c0d · 2ef408ff
Unverified Commit 2ef408ff authored Jan 28, 2026 by Schwinn Saereesitthipitak Committed by GitHub Jan 28, 2026
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

lib/gpu_memory_service/setup.py lib/gpu_memory_service/setup.py +2 -0

No files found.
--- a/lib/gpu_memory_service/setup.py
+++ b/lib/gpu_memory_service/setup.py
@@ -65,6 +65,7 @@ setup(
    # Package directory mapping: the current directory IS the gpu_memory_service package
    packages=[
        "gpu_memory_service",
+        "gpu_memory_service.cli",
        "gpu_memory_service.common",
        "gpu_memory_service.common.protocol",
        "gpu_memory_service.server",
@@ -75,6 +76,7 @@ setup(
    ],
    package_dir={
        "gpu_memory_service": ".",
+        "gpu_memory_service.cli": "cli",
        "gpu_memory_service.common": "common",
        "gpu_memory_service.common.protocol": "common/protocol",
        "gpu_memory_service.server": "server",