From e07236c22d7cb8fbda30d6d43e9a4601ecf9c7f5 Mon Sep 17 00:00:00 2001
From: Vyacheslav Matyukhin <me@berekuk.ru>
Date: Tue, 20 Sep 2022 20:54:26 +0400
Subject: [PATCH] fix scripts, support SAMPLE_COUNT

---
 packages/squiggle-lang/scripts/lib.mjs | 2 +-
 packages/squiggle-lang/scripts/run.mjs | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/squiggle-lang/scripts/lib.mjs b/packages/squiggle-lang/scripts/lib.mjs
index ffa1d84f..6f778769 100644
--- a/packages/squiggle-lang/scripts/lib.mjs
+++ b/packages/squiggle-lang/scripts/lib.mjs
@@ -13,7 +13,7 @@ export const measure = (cb, times = 1) => {
 export const red = (str) => `\x1b[31m${str}\x1b[0m`;
 export const green = (str) => `\x1b[32m${str}\x1b[0m`;
 
-export const run = (src, { output, sampleCount }) => {
+export const run = (src, { output, sampleCount } = {}) => {
   const project = SqProject.create();
   if (sampleCount) {
     project.setEnvironment({
diff --git a/packages/squiggle-lang/scripts/run.mjs b/packages/squiggle-lang/scripts/run.mjs
index bc17314b..6dde265f 100755
--- a/packages/squiggle-lang/scripts/run.mjs
+++ b/packages/squiggle-lang/scripts/run.mjs
@@ -7,4 +7,6 @@ if (!src) {
 }
 console.log(`Running ${src}`);
 
-run(src);
+const sampleCount = process.env.SAMPLE_COUNT;
+
+run(src, { output: true, sampleCount });