folder id :(

1 files changed, 6 insertions, 5 deletions

diff --git a/cli/app/search/json.py b/cli/app/search/json.py
index bf6bcb9..ced84df 100644
--- a/cli/app/search/json.py
+++ b/cli/app/search/json.py
@@ -55,9 +55,10 @@ def save_params_latent(fp_out_dir, tag):
   fp_out_fn = os.path.join(fp_out_dir, "params_latent.json")
   write_json(data, fp_out_fn)
 
-def make_params_dense(tag):
+def make_params_dense(tag, folder_id):
   return {
     "tag": tag,
+    "folder_id": folder_id,
     "decay_n": 2, 
     "features": True, 
     "clip": 1.0, 
@@ -100,10 +101,10 @@ def make_params_dense(tag):
     "max_batches": 0,
   }
 
-def params_dense_dict(tag):
-  return ParamsDict(make_params_dense(tag))
+def params_dense_dict(tag, folder_id=59):
+  return ParamsDict(make_params_dense(tag, folder_id))
 
-def save_params_dense(fp_out_dir, tag):
-  data = make_params_dense(tag)
+def save_params_dense(fp_out_dir, tag, folder_id=59):
+  data = make_params_dense(tag, folder_id)
   fp_out_fn = os.path.join(fp_out_dir, "params_dense.json")
   write_json(data, fp_out_fn)