Fix pylace dataset codebooks

promised-ai · Mar 20, 2024 · 45f1cc4 · 45f1cc4
1 parent a92ca9d
commit 45f1cc4
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/pylace/lace/resources/datasets/animals/codebook.yaml b/pylace/lace/resources/datasets/animals/codebook.yaml
@@ -1,10 +1,12 @@
 table_name: my_table
-state_alpha_prior:
-  shape: 1.0
-  rate: 1.0
-view_alpha_prior:
-  shape: 1.0
-  rate: 1.0
+state_prior_process: !dirichlet
+  alpha_prior:
+    shape: 1.0
+    rate: 1.0
+view_prior_process: !dirichlet
+  alpha_prior:
+    shape: 1.0
+    rate: 1.0
 col_metadata:
 - name: black
   coltype: !Categorical

diff --git a/pylace/lace/resources/datasets/satellites/codebook.yaml b/pylace/lace/resources/datasets/satellites/codebook.yaml
@@ -1,10 +1,12 @@
 table_name: my_data
-state_alpha_prior: !Gamma
-  shape: 1.0
-  rate: 1.0
-view_alpha_prior:
-  shape: 1.0
-  rate: 1.0
+state_prior_process: !dirichlet
+  alpha_prior:
+    shape: 1.0
+    rate: 1.0
+view_prior_process: !dirichlet
+  alpha_prior:
+    shape: 1.0
+    rate: 1.0
 col_metadata:
 - name: Country_of_Operator
   coltype: !Categorical