Fix lilac deployer for slashed datasets. (#1021)

databricks · Jan 4, 2024 · dd3071a · dd3071a
1 parent 9ebd89d
commit dd3071a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lilac_deployer/app.py b/lilac_deployer/app.py
@@ -91,7 +91,7 @@ def _next_button():
   if ds_builder:
     st.session_state.ds_loaded = True
     st.session_state.ds_error = None
-    st.session_state.ds_dataset_name = ds_builder.info.dataset_name
+    st.session_state.ds_dataset_name = hf_dataset_name
     st.session_state.ds_description = ds_builder.info.description
     st.session_state.ds_features = ds_builder.info.features
     st.session_state.ds_splits = ds_builder.info.splits