Merge pull request #99 from patrickfleith/bug/small-bug-fixes

Bug/small bug fixes

Author: patrickfleith

datafast/datasets.py

@@ -235,7 +235,7 @@ def push_to_hub(
         self,
         repo_id: str,
         token: str | None = None,
-        private: bool = False,
+        private: bool = True,
         commit_message: str | None = None,
         train_size: float | None = None,
         seed: int | None = None,

datafast/utils.py

@@ -120,7 +120,8 @@ def _get_mcq_specific_factors(config: MCQDatasetConfig) -> dict[str, int]:
 
 def _get_mcq_num_expected_rows(config: MCQDatasetConfig, llms: list[LLMProvider], source_data_num_rows: int) -> int:
     # factors = _get_mcq_specific_factors(config)  # Not specific factors
-    source_data_num_rows = min(source_data_num_rows, config.sample_count)
+    if config.sample_count is not None:
+        source_data_num_rows = min(source_data_num_rows, config.sample_count)
     num_llms = len(llms)
     if config.prompts is None:
         num_expanded_prompts = 1