- added support for passing command line arguments

Author: daslu

CHANGELOG.md

@@ -1,6 +1,9 @@
 # Change Log
 All notable changes to this project will be documented in this file. This change log follows the conventions of [keepachangelog.com](http://keepachangelog.com/).
 
+## [1-alpha3] - 2024-09-11
+- added support for passing command line arguments
+
 ## [1-alpha2] - 2024-07-30 
 - updated deps, added bababashka/process dependency
 - improved handling of shell commands

src/scicloj/cmdstan_clj/v1/api.clj

@@ -46,26 +46,35 @@
                (spit path code)
                path))))
 
-(defn model [code]
-  (let [path (-> code
-                 code->stan-path
-                 (str/replace #"\.stan$" ""))
-        ret (verbose-shell {:dir STAN_HOME}
-                           "make" path)]
-    (assoc ret
-           :path path)))
+(defn model
+  ([code]
+   (model code []))
+  ([code make-args]
+   (let [path (-> code
+                  code->stan-path
+                  (str/replace #"\.stan$" ""))
+         ret (apply verbose-shell
+                    {:dir STAN_HOME}
+                    "make"
+                    (conj make-args path))]
+     (assoc ret
+            :path path))))
 
 (defn sample
   ([model data]
    (sample model data {}))
-  ([model data {:keys [num-chains]}]
+  ([model data {:as options
+                :keys [num-chains]}]
    (let [data-path (:path (tempfiles/tempfile! ".json"))
          samples-path (:path (tempfiles/tempfile! ".csv"))
          _ (charred/write-json data-path data)
-         args [(when num-chains
-                 (str "num_chains=" num-chains))
-               "data" (str "file=" data-path)
-               "output" (str "file=" samples-path)]
+         args (concat (some->> options
+                               (map (fn [[k v]]
+                                      (format "%s=%s"
+                                              (-> k name (str/replace #"-" "_"))
+                                              (str v)))))
+                      ["data" (str "file=" data-path)
+                       "output" (str "file=" samples-path)])
          ret (->> args
                   (filter some?)
                   (apply verbose-shell {} (:path model) "sample"))