Update commandline interface (#6)

gsd-authors · Oct 17, 2023 · 47e1b56 · 47e1b56
1 parent a0fcb80
commit 47e1b56
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -54,7 +54,7 @@ $ hatch run test
 You can quickly estimate GSD parameters from a command line interface
 
 ```shell
-python3 -m gsd 0 12 13 4 0
+python3 -m gsd -c 1 2 3 4 5
 ```
 
-    GSDParams(psi=Array(2.6272388, dtype=float32), rho=Array(0.9041536, dtype=float32))
+    psi=3.6667 rho=0.6000
diff --git a/src/gsd/__main__.py b/src/gsd/__main__.py
@@ -4,12 +4,8 @@
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='GSD estimator')
-
-    parser.add_argument("response", nargs=5, type=int,
-                        metavar=("num1", "num2", "num3", "num4", "num5"),
-                        help="List of 5 counts")
-
+    parser.add_argument("-c", nargs=5, type=int,help="List of 5 counts", required=True)
     args = parser.parse_args()
 
-    hat,_ = fit_mle(data=jnp.asarray(args.response, dtype=jnp.float32))
-    print(hat)
+    hat,_ = fit_mle(data=jnp.asarray(args.o, dtype=jnp.float32))
+    print(f'psi={hat.psi:.4f} rho={hat.rho:.4f}')