src.tests.embedders.test_embedders.TestSentenceTransformersEmbedder ‑ test_instructor_embedding
src.tests.llms.test_llms.TestPetals ‑ test_petals_network
src.tests.steps.test_step_export.TestPublishToHFHub ‑ test_publish_dataset_to_hf_hub_single_train_split
src.tests.steps.test_step_export.TestPublishToHFHub ‑ test_publish_dataset_to_hf_hub_two_splits

src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_dataset[OutputDataset-False]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_dataset[OutputDatasetColumn-False]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_dataset[OutputIterableDataset-True]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_dataset[OutputIterableDatasetColumn-True]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_step[OutputDataset-False]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_step[OutputDatasetColumn-False]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_step[OutputIterableDataset-True]
src.tests.datasets.test_datasets.TestErrors ‑ test_invalid_args_step[OutputIterableDatasetColumn-True]
src.tests.datasets.test_datasets.TestErrors ‑ test_too_few_columns[OutputDatasetColumn-False]
src.tests.datasets.test_datasets.TestErrors ‑ test_too_few_columns[OutputIterableDatasetColumn-True]
src.tests.datasets.test_datasets.TestErrors ‑ test_too_many_columns[OutputDatasetColumn-False]
src.tests.datasets.test_datasets.TestErrors ‑ test_too_many_columns[OutputIterableDatasetColumn-True]
src.tests.datasets.test_datasets.TestFunctionality ‑ test_dataset
src.tests.datasets.test_datasets.TestFunctionality ‑ test_dataset_indexing
src.tests.datasets.test_datasets.TestFunctionality ‑ test_dataset_pickled
src.tests.datasets.test_datasets.TestFunctionality ‑ test_iterable_dataset
src.tests.datasets.test_datasets.TestFunctionality ‑ test_iterable_dataset_indexing
src.tests.datasets.test_datasets.TestFunctionality ‑ test_iterable_dataset_pickled
src.tests.datasets.test_utils.TestErrors ‑ test_dataset_zip_no_args
src.tests.datasets.test_utils.TestErrors ‑ test_iterable_dataset_zip_no_args
src.tests.datasets.test_utils.TestFunctionality ‑ test_dataset_zip
src.tests.datasets.test_utils.TestFunctionality ‑ test_get_column_names
src.tests.datasets.test_utils.TestFunctionality ‑ test_iterable_dataset_zip
src.tests.embedders.test_embedders.TestOpenAIEmbedder ‑ test_init
src.tests.embedders.test_embedders.TestOpenAIEmbedder ‑ test_metadata
src.tests.embedders.test_embedders.TestOpenAIEmbedder ‑ test_run
src.tests.embedders.test_embedders.TestParallelEmbedder ‑ test_run
src.tests.embedders.test_embedders.TestSentenceTransformersEmbedder ‑ test_init
src.tests.embedders.test_embedders.TestSentenceTransformersEmbedder ‑ test_instructor_embedding
src.tests.embedders.test_embedders.TestSentenceTransformersEmbedder ‑ test_metadata
src.tests.embedders.test_embedders.TestSentenceTransformersEmbedder ‑ test_run
src.tests.embedders.test_embedders.TestTogetherEmbedder ‑ test_init
src.tests.embedders.test_embedders.TestTogetherEmbedder ‑ test_metadata
src.tests.embedders.test_embedders.TestTogetherEmbedder ‑ test_run
src.tests.llms.test_llms.TestAI21 ‑ test_init
src.tests.llms.test_llms.TestAI21 ‑ test_run
src.tests.llms.test_llms.TestAnthropic ‑ test_init
src.tests.llms.test_llms.TestAnthropic ‑ test_run
src.tests.llms.test_llms.TestBedrock ‑ test_init
src.tests.llms.test_llms.TestBedrock ‑ test_run
src.tests.llms.test_llms.TestCTransformers ‑ test_count_tokens
src.tests.llms.test_llms.TestCTransformers ‑ test_get_max_context_length
src.tests.llms.test_llms.TestCTransformers ‑ test_init
src.tests.llms.test_llms.TestCTransformers ‑ test_metadata
src.tests.llms.test_llms.TestCTransformers ‑ test_run
src.tests.llms.test_llms.TestCohere ‑ test_init
src.tests.llms.test_llms.TestCohere ‑ test_run
src.tests.llms.test_llms.TestHFAPIEndpoint ‑ test_init
src.tests.llms.test_llms.TestHFAPIEndpoint ‑ test_metadata
src.tests.llms.test_llms.TestHFAPIEndpoint ‑ test_run
src.tests.llms.test_llms.TestHFTransformers ‑ test_adapter_metadata
src.tests.llms.test_llms.TestHFTransformers ‑ test_count_tokens
src.tests.llms.test_llms.TestHFTransformers ‑ test_get_max_context_length
src.tests.llms.test_llms.TestHFTransformers ‑ test_init
src.tests.llms.test_llms.TestHFTransformers ‑ test_length_errors
src.tests.llms.test_llms.TestHFTransformers ‑ test_metadata
src.tests.llms.test_llms.TestHFTransformers ‑ test_non_popular_chat_model
src.tests.llms.test_llms.TestHFTransformers ‑ test_popular_chat_model
src.tests.llms.test_llms.TestHFTransformers ‑ test_run
src.tests.llms.test_llms.TestHFTransformers ‑ test_run_adapter
src.tests.llms.test_llms.TestLLM ‑ test_adaptive_batch_size_and_batch_scheduler
src.tests.llms.test_llms.TestLLM ‑ test_batch_size_causes_oom_error
src.tests.llms.test_llms.TestLLM ‑ test_cache_created
src.tests.llms.test_llms.TestLLM ‑ test_cache_created_in_custom_location
src.tests.llms.test_llms.TestLLM ‑ test_cached_tokenizer_during_adaptive_batch_retry
src.tests.llms.test_llms.TestLLM ‑ test_chat_prompt_templates
src.tests.llms.test_llms.TestLLM ‑ test_check_max_new_tokens_possible
src.tests.llms.test_llms.TestLLM ‑ test_check_temperature_and_top_p
src.tests.llms.test_llms.TestLLM ‑ test_disable_adaptive_batch_size
src.tests.llms.test_llms.TestLLM ‑ test_disable_batch_scheduler
src.tests.llms.test_llms.TestLLM ‑ test_error_in_run_propagates
src.tests.llms.test_llms.TestLLM ‑ test_format_prompt
src.tests.llms.test_llms.TestLLM ‑ test_format_prompt_error_instruction_too_large
src.tests.llms.test_llms.TestLLM ‑ test_format_prompt_error_instruction_too_large_with_in_context_examples
src.tests.llms.test_llms.TestLLM ‑ test_import
src.tests.llms.test_llms.TestLLM ‑ test_is_primitive
src.tests.llms.test_llms.TestLLM ‑ test_llm_lock
src.tests.llms.test_llms.TestLLM ‑ test_progress_logging
src.tests.llms.test_llms.TestLLM ‑ test_repr
src.tests.llms.test_llms.TestLLM ‑ test_run_with_invalid_batch_size
src.tests.llms.test_llms.TestLLM ‑ test_run_with_no_prompts
src.tests.llms.test_llms.TestLLM ‑ test_run_with_repeated_prompts_with_cache
src.tests.llms.test_llms.TestLLM ‑ test_run_with_repeated_prompts_without_cache
src.tests.llms.test_llms.TestLLM ‑ test_too_many_in_context_examples
src.tests.llms.test_llms.TestLiteLLM ‑ test_count_tokens
src.tests.llms.test_llms.TestLiteLLM ‑ test_get_max_context_length
src.tests.llms.test_llms.TestLiteLLM ‑ test_run
src.tests.llms.test_llms.TestMistralAI ‑ test_count_tokens
src.tests.llms.test_llms.TestMistralAI ‑ test_get_max_context_length
src.tests.llms.test_llms.TestMistralAI ‑ test_init
src.tests.llms.test_llms.TestMistralAI ‑ test_metadata
src.tests.llms.test_llms.TestMistralAI ‑ test_run
src.tests.llms.test_llms.TestOpenAI ‑ test_count_tokens
src.tests.llms.test_llms.TestOpenAI ‑ test_get_max_context_length
src.tests.llms.test_llms.TestOpenAI ‑ test_get_max_output_length
src.tests.llms.test_llms.TestOpenAI ‑ test_init
src.tests.llms.test_llms.TestOpenAI ‑ test_length_errors
src.tests.llms.test_llms.TestOpenAI ‑ test_metadata
src.tests.llms.test_llms.TestOpenAI ‑ test_run
src.tests.llms.test_llms.TestOpenAIAssistant ‑ test_init
src.tests.llms.test_llms.TestOpenAIAssistant ‑ test_run
src.tests.llms.test_llms.TestPaLM ‑ test_init
src.tests.llms.test_llms.TestPaLM ‑ test_run
src.tests.llms.test_llms.TestParallelLLM ‑ test_count_tokens
src.tests.llms.test_llms.TestParallelLLM ‑ test_format_prompt
src.tests.llms.test_llms.TestParallelLLM ‑ test_get_max_context_length
src.tests.llms.test_llms.TestParallelLLM ‑ test_init
src.tests.llms.test_llms.TestParallelLLM ‑ test_metadata
src.tests.llms.test_llms.TestParallelLLM ‑ test_parallel_llm_lock
src.tests.llms.test_llms.TestParallelLLM ‑ test_run_empty
src.tests.llms.test_llms.TestParallelLLM ‑ test_run_large
src.tests.llms.test_llms.TestParallelLLM ‑ test_run_return_generator
src.tests.llms.test_llms.TestParallelLLM ‑ test_run_small
src.tests.llms.test_llms.TestPetals ‑ test_adaptive_batch_size
src.tests.llms.test_llms.TestPetals ‑ test_init
src.tests.llms.test_llms.TestPetals ‑ test_metadata
src.tests.llms.test_llms.TestPetals ‑ test_petals_network
src.tests.llms.test_llms.TestPetals ‑ test_run
src.tests.llms.test_llms.TestTogether ‑ test_count_tokens
src.tests.llms.test_llms.TestTogether ‑ test_get_max_context_length
src.tests.llms.test_llms.TestTogether ‑ test_init
src.tests.llms.test_llms.TestTogether ‑ test_metadata
src.tests.llms.test_llms.TestTogether ‑ test_run
src.tests.llms.test_llms.TestTogether ‑ test_warnings
src.tests.llms.test_llms.TestVertexAI ‑ test_init
src.tests.llms.test_llms.TestVertexAI ‑ test_run
src.tests.retrievers.test_retrievers.TestEmbeddingRetriever ‑ test_in_memory_index
src.tests.retrievers.test_retrievers.TestEmbeddingRetriever ‑ test_index
src.tests.retrievers.test_retrievers.TestEmbeddingRetriever ‑ test_init
src.tests.retrievers.test_retrievers.TestEmbeddingRetriever ‑ test_metadata
src.tests.retrievers.test_retrievers.TestEmbeddingRetriever ‑ test_run
src.tests.retrievers.test_retrievers.TestParallelRetriever ‑ test_run
src.tests.steps.prompt.test_prompt.TestDataFromAttributedPrompt ‑ test_data_from_attributed_prompt
src.tests.steps.prompt.test_prompt.TestDataFromAttributedPrompt ‑ test_data_from_attributed_prompt_with_list_of_combinations
src.tests.steps.prompt.test_prompt.TestDataFromPrompt ‑ test_data_from_prompt
src.tests.steps.prompt.test_prompt.TestDataFromPrompt ‑ test_data_from_prompt_with_seed
src.tests.steps.prompt.test_prompt.TestFewShotPrompt ‑ test_few_shot_prompt
src.tests.steps.prompt.test_prompt.TestFewShotPrompt ‑ test_few_shot_prompt_custom
src.tests.steps.prompt.test_prompt.TestFewShotPromptWithRetrieval ‑ test_few_shot_prompt_with_retrieval
src.tests.steps.prompt.test_prompt.TestFilterWithPrompt ‑ test_filter_with_prompt
src.tests.steps.prompt.test_prompt.TestJudgeGenerationPairsWithPrompt ‑ test_judge_generation_pairs_with_prompt
src.tests.steps.prompt.test_prompt.TestJudgePairsWithPrompt ‑ test_judge_pairs_with_prompt
src.tests.steps.prompt.test_prompt.TestProcessWithPrompt ‑ test_process_with_prompt
src.tests.steps.prompt.test_prompt.TestPrompt ‑ test_prompt
src.tests.steps.prompt.test_prompt.TestPromptBase ‑ test_prompt_base
src.tests.steps.prompt.test_prompt.TestPromptBase ‑ test_prompt_base_lazy
src.tests.steps.prompt.test_prompt.TestPromptBase ‑ test_prompt_logging
src.tests.steps.prompt.test_prompt.TestPromptBase ‑ test_prompt_logging_iterable
src.tests.steps.prompt.test_prompt.TestRAGPrompt ‑ test_rag_prompt
src.tests.steps.prompt.test_prompt.TestRankWithPrompt ‑ test_rank_with_prompt
src.tests.steps.tasks.test_tasks.TestCosineSimilarity ‑ test_cosine_sim_between_embeddings
src.tests.steps.tasks.test_tasks.TestCosineSimilarity ‑ test_cosine_sim_between_texts
src.tests.steps.tasks.test_tasks.TestCosineSimilarity ‑ test_cosine_sim_on_multiple_devices
src.tests.steps.tasks.test_tasks.TestEmbed ‑ test_embed
src.tests.steps.tasks.test_tasks.TestRetrieve ‑ test_embed
src.tests.steps.tasks.test_tasks.TestRunTaskModel ‑ test_run_task_model
src.tests.steps.test_data_sources.TestCSVDataSource ‑ test_from_csv_file
src.tests.steps.test_data_sources.TestCSVDataSource ‑ test_from_csv_file_splits_error
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_dataset
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_dataset_dict_error
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_dict
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_generator
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_generator_warning
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_iterable_dataset
src.tests.steps.test_data_sources.TestDataSource ‑ test_from_list
src.tests.steps.test_data_sources.TestHFDatasetDataSource ‑ test_from_hf_dataset_folder
src.tests.steps.test_data_sources.TestHFDatasetDataSource ‑ test_from_hf_dataset_folder_splits_error
src.tests.steps.test_data_sources.TestHFHubDataSource ‑ test_from_hf_hub
src.tests.steps.test_data_sources.TestHFHubDataSource ‑ test_from_hf_hub_splits_error
src.tests.steps.test_data_sources.TestJSONDataSource ‑ test_from_json_file
src.tests.steps.test_data_sources.TestJSONDataSource ‑ test_from_json_file_splits_error
src.tests.steps.test_data_sources.TestTextDataSource ‑ test_from_text_file
src.tests.steps.test_data_sources.TestTextDataSource ‑ test_from_text_file_splits_error
src.tests.steps.test_step.TestErrors ‑ test_empty_name
src.tests.steps.test_step.TestErrors ‑ test_init_cannot_be_overridden
src.tests.steps.test_step.TestErrors ‑ test_invalid_arguments_to_register_functions
src.tests.steps.test_step.TestErrors ‑ test_invalid_initialization
src.tests.steps.test_step.TestErrors ‑ test_no_outputs_registered
src.tests.steps.test_step.TestErrors ‑ test_register_functions_run_after_initialization
src.tests.steps.test_step.TestErrors ‑ test_step_run_not_implemented
src.tests.steps.test_step.TestErrors ‑ test_step_setup_not_implemented
src.tests.steps.test_step.TestFunctionality ‑ test_data_card
src.tests.steps.test_step.TestFunctionality ‑ test_head
src.tests.steps.test_step.TestFunctionality ‑ test_head_shuffle
src.tests.steps.test_step.TestFunctionality ‑ test_new_step
src.tests.steps.test_step.TestFunctionality ‑ test_outputs_renames_columns
src.tests.steps.test_step.TestFunctionality ‑ test_step_help_string
src.tests.steps.test_step.TestFunctionality ‑ test_step_str_repr
src.tests.steps.test_step.TestSuperStep ‑ test_non_superstep
src.tests.steps.test_step.TestSuperStep ‑ test_superstep
src.tests.steps.test_step.TestSuperStep ‑ test_superstep_background
src.tests.steps.test_step.TestSuperStep ‑ test_superstep_concurrent
src.tests.steps.test_step.TestSuperStep ‑ test_superstep_data_card_propagates
src.tests.steps.test_step_background.TestBackground ‑ test_can_get_background_progress_in_foreground
src.tests.steps.test_step_background.TestBackground ‑ test_can_get_background_progress_rows_in_foreground
src.tests.steps.test_step_background.TestBackground ‑ test_concurrent_step_operations
src.tests.steps.test_step_background.TestBackground ‑ test_generator_step_with_background_runs_in_background_process
src.tests.steps.test_step_background.TestBackground ‑ test_step_operation_on_backgrounded_step
src.tests.steps.test_step_background.TestBackground ‑ test_step_with_background_runs_in_background_process
src.tests.steps.test_step_background.TestBackground ‑ test_step_without_background_runs_in_same_process
src.tests.steps.test_step_background.TestErrors ‑ test_concurrent_invalid_args
src.tests.steps.test_step_background.TestErrors ‑ test_wait_invalid_args
src.tests.steps.test_step_export.TestError ‑ test_export_dataset_to_dict_two_splits_by_row_wrong_proportions
src.tests.steps.test_step_export.TestError ‑ test_export_dataset_to_dict_two_splits_wrong_proportions
src.tests.steps.test_step_export.TestExportToCSV ‑ test_export_dataset_to_list_single_train_split
src.tests.steps.test_step_export.TestExportToCSV ‑ test_export_dataset_to_list_two_splits
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_dataset_to_dict_no_split
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_dataset_to_dict_single_other_split
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_dataset_to_dict_single_train_split
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_dataset_to_dict_three_splits
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_dataset_to_dict_two_splits
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_dataset_to_dict_two_splits_by_row
src.tests.steps.test_step_export.TestExportToDict ‑ test_export_iterable_dataset_to_dict_three_splits
src.tests.steps.test_step_export.TestExportToHFDataset ‑ test_export_dataset_to_hf_dataset_single_train_split
src.tests.steps.test_step_export.TestExportToHFDataset ‑ test_export_dataset_to_hf_dataset_two_splits
src.tests.steps.test_step_export.TestExportToJSON ‑ test_export_dataset_to_list_single_train_split
src.tests.steps.test_step_export.TestExportToJSON ‑ test_export_dataset_to_list_two_splits
src.tests.steps.test_step_export.TestExportToList ‑ test_export_dataset_to_list_single_train_split
src.tests.steps.test_step_export.TestExportToList ‑ test_export_dataset_to_list_two_splits
src.tests.steps.test_step_export.TestPickle ‑ test_export_dataset_to_dict_pickled
src.tests.steps.test_step_export.TestPickle ‑ test_export_dataset_to_hf_dataset_pickled
src.tests.steps.test_step_export.TestPickle ‑ test_export_dataset_to_list_pickled
src.tests.steps.test_step_export.TestPublishToHFHub ‑ test_code_implementation
src.tests.steps.test_step_export.TestPublishToHFHub ‑ test_publish_dataset_to_hf_hub_single_train_split
src.tests.steps.test_step_export.TestPublishToHFHub ‑ test_publish_dataset_to_hf_hub_two_splits
src.tests.steps.test_step_operations.TestAddItem ‑ test_add_item_dataset
src.tests.steps.test_step_operations.TestAddItem ‑ test_add_item_iterable_dataset
src.tests.steps.test_step_operations.TestConcat ‑ test_concat
src.tests.steps.test_step_operations.TestConcat ‑ test_concat_invalid_args
src.tests.steps.test_step_operations.TestConcat ‑ test_concat_lazy
src.tests.steps.test_step_operations.TestConcat ‑ test_concat_lazy_with_no_num_rows
src.tests.steps.test_step_operations.TestConcat ‑ test_concat_with_different_column_names
src.tests.steps.test_step_operations.TestCopy ‑ test_copy_dataset
src.tests.steps.test_step_operations.TestCopy ‑ test_copy_dataset_lazy
src.tests.steps.test_step_operations.TestCopy ‑ test_copy_iterable_dataset
src.tests.steps.test_step_operations.TestCopy ‑ test_copy_iterable_dataset_not_lazy
src.tests.steps.test_step_operations.TestFilter ‑ test_filter_on_dataset
src.tests.steps.test_step_operations.TestFilter ‑ test_filter_on_dataset_lazy
src.tests.steps.test_step_operations.TestFilter ‑ test_filter_on_iterable_dataset
src.tests.steps.test_step_operations.TestFilter ‑ test_filter_to_empty_dataset
src.tests.steps.test_step_operations.TestFilter ‑ test_filter_to_empty_dataset_lazy
src.tests.steps.test_step_operations.TestFilter ‑ test_filter_with_total_num_rows
src.tests.steps.test_step_operations.TestMap ‑ test_map_add_remove_columns
src.tests.steps.test_step_operations.TestMap ‑ test_map_empty_generator
src.tests.steps.test_step_operations.TestMap ‑ test_map_lazy_pickle_error
src.tests.steps.test_step_operations.TestMap ‑ test_map_on_dataset
src.tests.steps.test_step_operations.TestMap ‑ test_map_on_iterable_dataset
src.tests.steps.test_step_operations.TestMap ‑ test_map_pickle_error
src.tests.steps.test_step_operations.TestMap ‑ test_map_with_total_num_rows
src.tests.steps.test_step_operations.TestRemoveColumns ‑ test_remove_columns_dataset
src.tests.steps.test_step_operations.TestRemoveColumns ‑ test_remove_columns_iterable_dataset
src.tests.steps.test_step_operations.TestRenameColumn ‑ test_rename_column_dataset
src.tests.steps.test_step_operations.TestRenameColumn ‑ test_rename_column_iterable_dataset
src.tests.steps.test_step_operations.TestRenameColumn ‑ test_rename_with_same_name
src.tests.steps.test_step_operations.TestRenameColumns ‑ test_rename_columns_dataset
src.tests.steps.test_step_operations.TestRenameColumns ‑ test_rename_columns_iterable_dataset
src.tests.steps.test_step_operations.TestRenameColumns ‑ test_rename_columns_with_same_name
src.tests.steps.test_step_operations.TestReverse ‑ test_reverse_dataset
src.tests.steps.test_step_operations.TestReverse ‑ test_reverse_iterable_dataset
src.tests.steps.test_step_operations.TestSave ‑ test_save_num_shards
src.tests.steps.test_step_operations.TestSave ‑ test_save_on_dataset
src.tests.steps.test_step_operations.TestSave ‑ test_save_on_iterable_dataset
src.tests.steps.test_step_operations.TestSave ‑ test_save_step_naming
src.tests.steps.test_step_operations.TestSelect ‑ test_select_dataset
src.tests.steps.test_step_operations.TestSelect ‑ test_select_iterable_dataset
src.tests.steps.test_step_operations.TestSelectColumns ‑ test_select_columns_dataset
src.tests.steps.test_step_operations.TestSelectColumns ‑ test_select_columns_iterable_dataset
src.tests.steps.test_step_operations.TestShard ‑ test_shard_dataset
src.tests.steps.test_step_operations.TestShard ‑ test_shard_dataset_contiguous
src.tests.steps.test_step_operations.TestShard ‑ test_shard_iterable_dataset
src.tests.steps.test_step_operations.TestShuffle ‑ test_dataset
src.tests.steps.test_step_operations.TestShuffle ‑ test_dataset_lazy
src.tests.steps.test_step_operations.TestShuffle ‑ test_iterable_dataset
src.tests.steps.test_step_operations.TestShuffle ‑ test_iterable_dataset_lazy
src.tests.steps.test_step_operations.TestSkip ‑ test_skip_dataset
src.tests.steps.test_step_operations.TestSkip ‑ test_skip_dataset_over_length
src.tests.steps.test_step_operations.TestSkip ‑ test_skip_iterable_dataset
src.tests.steps.test_step_operations.TestSkip ‑ test_skip_iterable_dataset_over_length
src.tests.steps.test_step_operations.TestSort ‑ test_sort_dataset
src.tests.steps.test_step_operations.TestSort ‑ test_sort_iterable_dataset
src.tests.steps.test_step_operations.TestSplits ‑ test_splits_dataset
src.tests.steps.test_step_operations.TestSplits ‑ test_splits_iterable_dataset
src.tests.steps.test_step_operations.TestTake ‑ test_take_dataset
src.tests.steps.test_step_operations.TestTake ‑ test_take_iterable_dataset
src.tests.steps.test_step_operations.TestTake ‑ test_take_iterable_dataset_over_length
src.tests.steps.test_step_operations.TestZipped ‑ test_zipped
src.tests.steps.test_step_operations.TestZipped ‑ test_zipped_invalid_args
src.tests.steps.test_step_operations.TestZipped ‑ test_zipped_lazy
src.tests.steps.test_step_operations.TestZipped ‑ test_zipped_lazy_with_no_num_rows
src.tests.steps.test_step_operations.TestZipped ‑ test_zipped_with_different_lengths_error
src.tests.steps.test_step_output.TestColumn ‑ test_output_dict_of_OutputDatasetColumn
src.tests.steps.test_step_output.TestColumn ‑ test_output_dict_of_OutputIterableDatasetColumn
src.tests.steps.test_step_output.TestColumn ‑ test_output_list_of_OutputDatasetColumn
src.tests.steps.test_step_output.TestColumn ‑ test_output_single_OutputDatasetColumn
src.tests.steps.test_step_output.TestColumn ‑ test_output_single_OutputIterableDatasetColumn
src.tests.steps.test_step_output.TestColumn ‑ test_output_tuple_of_OutputDatasetColumn
src.tests.steps.test_step_output.TestColumn ‑ test_output_tuple_of_OutputIterableDatasetColumn
src.tests.steps.test_step_output.TestDataset ‑ test_output_OutputDataset
src.tests.steps.test_step_output.TestDataset ‑ test_output_OutputIterableDataset
src.tests.steps.test_step_output.TestDataset ‑ test_output_OutputIterableDataset_batched
src.tests.steps.test_step_output.TestDataset ‑ test_output_list_of_OutputDatasets
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_dict_of_list_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_generator_function_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_generator_function_of_dict_batched_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_generator_function_of_tuple_batched_column_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_generator_function_of_tuple_batched_row_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_iterator_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_list_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_single_empty
src.tests.steps.test_step_output.TestEmptyOutput ‑ test_output_tuple_of_list_empty
src.tests.steps.test_step_output.TestErrors ‑ test_access_output_before_step_is_run
src.tests.steps.test_step_output.TestErrors ‑ test_access_output_while_step_is_running
src.tests.steps.test_step_output.TestErrors ‑ test_exceptions_propagate
src.tests.steps.test_step_output.TestErrors ‑ test_output_dataset_with_wrong_number_of_columns
src.tests.steps.test_step_output.TestErrors ‑ test_output_dict_with_wrong_keys
src.tests.steps.test_step_output.TestErrors ‑ test_output_generator_function_of_dict_with_wrong_keys
src.tests.steps.test_step_output.TestErrors ‑ test_output_generator_function_of_list_of_dict_batched_with_wrong_keys
src.tests.steps.test_step_output.TestErrors ‑ test_output_generator_function_of_list_with_wrong_number_of_outputs
src.tests.steps.test_step_output.TestErrors ‑ test_output_generator_function_of_tuple_batched_column_with_wrong_number_of_outputs
src.tests.steps.test_step_output.TestErrors ‑ test_output_generator_function_of_tuple_with_wrong_number_of_outputs
src.tests.steps.test_step_output.TestErrors ‑ test_output_invalid_type
src.tests.steps.test_step_output.TestErrors ‑ test_output_list_of_dicts_with_wrong_keys
src.tests.steps.test_step_output.TestErrors ‑ test_output_list_with_wrong_number_of_outputs
src.tests.steps.test_step_output.TestErrors ‑ test_output_tuple_with_wrong_number_of_outputs
src.tests.steps.test_step_output.TestErrors ‑ test_output_twice
src.tests.steps.test_step_output.TestErrors ‑ test_pickle_warning
src.tests.steps.test_step_output.TestErrors ‑ test_total_num_rows_warnings
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_dataset
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_dict_of_generator_function
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_dict_of_generator_function_batched
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_dict_of_iterator
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_dict_of_list
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_dict
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_dict_batched
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_iterator_row
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_list_row
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_list_row_batched_column
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_list_row_batched_column_ambiguous
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_list_row_batched_column_iterator
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_list_row_batched_row
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_list_row_batched_row_iterator
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_tuple
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_tuple_batched_column
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_generator_function_of_tuple_batched_row
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_iterable_dataset
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_iterable_dataset_batched
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_datasets
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_generator_function
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_generator_function_batched
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_iterable_datasets
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_list_column
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_list_row
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_list_of_tuple
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_single_dict
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_single_list
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_single_tuple
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_tuple_of_datasets
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_tuple_of_generator_function
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_tuple_of_generator_function_batched
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_tuple_of_iterator
src.tests.steps.test_step_output.TestMultipleOutput ‑ test_output_tuple_of_list
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress_batched
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress_rows
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress_rows_step_operation_map
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress_rows_step_operation_shuffle
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress_step_operation_map
src.tests.steps.test_step_output.TestProgress ‑ test_auto_progress_step_operation_shuffle
src.tests.steps.test_step_output.TestProgress ‑ test_before_output
src.tests.steps.test_step_output.TestProgress ‑ test_get_progress_string
src.tests.steps.test_step_output.TestProgress ‑ test_progress_after_output_dataset
src.tests.steps.test_step_output.TestProgress ‑ test_progress_after_output_iterable_dataset
src.tests.steps.test_step_output.TestProgress ‑ test_progress_after_output_list
src.tests.steps.test_step_output.TestProgress ‑ test_progress_after_output_single
src.tests.steps.test_step_output.TestProgress ‑ test_progress_after_output_tuple_of_list
src.tests.steps.test_step_output.TestProgress ‑ test_progress_is_1_after__output_is_dataset
src.tests.steps.test_step_output.TestProgress ‑ test_progress_monotonic_increases
src.tests.steps.test_step_output.TestProgress ‑ test_progress_range_higher
src.tests.steps.test_step_output.TestProgress ‑ test_progress_range_lower
src.tests.steps.test_step_output.TestProgress ‑ test_set_progress
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_dataset
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_dict_of_generator_function
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_dict_of_generator_function_batched
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_dict_of_generator_function_batched_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_dict_of_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_dict_of_list
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_dict
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_dict_batched
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_dict_batched_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_of_dict_batched
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_of_dict_batched_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_row
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_row_batched_column
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_row_batched_column_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_row_batched_row
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_list_row_batched_row_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_single
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_tuple
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_tuple_batched_column
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_tuple_batched_column_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_tuple_batched_row
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_generator_function_of_tuple_batched_row_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_iterable_dataset
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_iterable_dataset_batched
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_datasets
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_dicts
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_dicts_matching
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_generator_function
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_generator_function_batched
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_generator_function_batched_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_list_column
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_list_row
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_list_of_tuple
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_single
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_single_dict
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_single_list
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_single_tuple
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_tuple_of_datasets
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_tuple_of_generator_function
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_tuple_of_generator_function_batched
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_tuple_of_generator_function_batched_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_tuple_of_iterator
src.tests.steps.test_step_output.TestSingleOutput ‑ test_output_tuple_of_list
src.tests.steps.test_step_output.TestTypes ‑ test_dict_and_none
src.tests.steps.test_step_output.TestTypes ‑ test_dict_with_different_keys
src.tests.steps.test_step_output.TestTypes ‑ test_dict_with_no_keys
src.tests.steps.test_step_output.TestTypes ‑ test_features
src.tests.steps.test_step_output.TestTypes ‑ test_func
src.tests.steps.test_step_output.TestTypes ‑ test_int_and_float
src.tests.steps.test_step_output.TestTypes ‑ test_int_and_none
src.tests.steps.test_step_output.TestTypes ‑ test_int_and_str
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_dict_different_shape
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_features
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_features_empty
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_int_and_str
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_int_and_str_save
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_list_with_different_lengths
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_list_with_int_and_str
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_list_with_str_and_int
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_non_dict_and_dict
src.tests.steps.test_step_output.TestTypes ‑ test_iterable_dataset_str_and_int
src.tests.steps.test_step_output.TestTypes ‑ test_list_and_non_list
src.tests.steps.test_step_output.TestTypes ‑ test_list_with_different_lengths
src.tests.steps.test_step_output.TestTypes ‑ test_list_with_int_and_str
src.tests.steps.test_step_output.TestTypes ‑ test_list_with_no_elements
src.tests.steps.test_step_output.TestTypes ‑ test_list_with_str_and_int
src.tests.steps.test_step_output.TestTypes ‑ test_non_dict_and_dict
src.tests.steps.test_step_output.TestTypes ‑ test_pickle
src.tests.steps.test_step_output.TestTypes ‑ test_pickle_iterable_dataset
src.tests.steps.test_step_output.TestTypes ‑ test_str_and_datetime
src.tests.steps.test_step_output.TestTypes ‑ test_str_and_func
src.tests.steps.test_step_output.TestTypes ‑ test_str_and_int
src.tests.steps.test_step_output.TestTypes ‑ test_unpickle
src.tests.task_models.test_task_models.TestHFClassificationTaskModel ‑ test_init
src.tests.task_models.test_task_models.TestHFClassificationTaskModel ‑ test_metadata
src.tests.task_models.test_task_models.TestHFClassificationTaskModel ‑ test_peft
src.tests.task_models.test_task_models.TestHFClassificationTaskModel ‑ test_run
src.tests.test_cli.TestCli ‑ test_help
src.tests.test_datadreamer.TestErrors ‑ test_nested
src.tests.test_datadreamer.TestErrors ‑ test_path_is_to_file
src.tests.test_datadreamer.TestErrors ‑ test_steps_with_same_name
src.tests.test_datadreamer.TestFunctionality ‑ test_creates_folder
src.tests.test_datadreamer.TestFunctionality ‑ test_ctx_clears
src.tests.test_datadreamer.TestFunctionality ‑ test_data_card_propagates
src.tests.test_datadreamer.TestFunctionality ‑ test_in_memory
src.tests.test_datadreamer.TestFunctionality ‑ test_logging
src.tests.test_datadreamer.TestFunctionality ‑ test_num_shards
src.tests.test_datadreamer.TestFunctionality ‑ test_saves_pickled_dataset
src.tests.test_datadreamer.TestFunctionality ‑ test_step_creates_folder_and_saves_dataset
src.tests.test_datadreamer.TestFunctionality ‑ test_step_creates_run_output_folder
src.tests.test_datadreamer.TestFunctionality ‑ test_step_does_not_save_iterable_dataset
src.tests.test_datadreamer.TestFunctionality ‑ test_step_does_save_iterable_dataset
src.tests.test_datadreamer.TestFunctionality ‑ test_step_force
src.tests.test_datadreamer.TestFunctionality ‑ test_step_logging
src.tests.test_datadreamer.TestFunctionality ‑ test_steps_added_to_ctx
src.tests.test_package.TestPackage ‑ test_mypy
src.tests.test_package.TestPackage ‑ test_python_version
src.tests.test_package.TestPackage ‑ test_running_in_pytest
src.tests.test_package.TestPackage ‑ test_version
src.tests.trainers.test_trainers.TestTrainHFBase ‑ test_hf_chat_template
src.tests.trainers.test_trainers.TestTrainHFBase ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainHFBase ‑ test_model_card
src.tests.trainers.test_trainers.TestTrainHFBase ‑ test_peft_resume
src.tests.trainers.test_trainers.TestTrainHFBase ‑ test_resume
src.tests.trainers.test_trainers.TestTrainHFBase ‑ test_validate_peft_config
src.tests.trainers.test_trainers.TestTrainHFClassifier ‑ test_init
src.tests.trainers.test_trainers.TestTrainHFClassifier ‑ test_int_labels
src.tests.trainers.test_trainers.TestTrainHFClassifier ‑ test_multi_target
src.tests.trainers.test_trainers.TestTrainHFClassifier ‑ test_peft
src.tests.trainers.test_trainers.TestTrainHFClassifier ‑ test_peft_multi_target
src.tests.trainers.test_trainers.TestTrainHFDPO ‑ test_causal
src.tests.trainers.test_trainers.TestTrainHFDPO ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainHFDPO ‑ test_peft
src.tests.trainers.test_trainers.TestTrainHFDPO ‑ test_seq2seq
src.tests.trainers.test_trainers.TestTrainHFFineTune ‑ test_causal
src.tests.trainers.test_trainers.TestTrainHFFineTune ‑ test_instruction_tune
src.tests.trainers.test_trainers.TestTrainHFFineTune ‑ test_seq2seq
src.tests.trainers.test_trainers.TestTrainHFPPO ‑ test_causal
src.tests.trainers.test_trainers.TestTrainHFPPO ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainHFPPO ‑ test_peft
src.tests.trainers.test_trainers.TestTrainHFPPO ‑ test_resume
src.tests.trainers.test_trainers.TestTrainHFPPO ‑ test_reward_model_from_trainer
src.tests.trainers.test_trainers.TestTrainHFPPO ‑ test_seq2seq
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_do_not_use_train
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_pairs_and_scores_causal
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_pairs_causal
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_pairs_peft
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_pairs_seq2seq
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_scores_causal
src.tests.trainers.test_trainers.TestTrainHFRewardModel ‑ test_scores_peft
src.tests.trainers.test_trainers.TestTrainOpenAIFineTune ‑ test_finetune[False-babbage-002]
src.tests.trainers.test_trainers.TestTrainOpenAIFineTune ‑ test_finetune[True-gpt-3.5-turbo-1106]
src.tests.trainers.test_trainers.TestTrainOpenAIFineTune ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainOpenAIFineTune ‑ test_truncate_error
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_do_not_use_train
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_labeled_pairs
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_peft
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_peft_resume
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_positive_pairs
src.tests.trainers.test_trainers.TestTrainSentenceTransformer ‑ test_triplets
src.tests.trainers.test_trainers.TestTrainSetFitClassifier ‑ test_init
src.tests.trainers.test_trainers.TestTrainSetFitClassifier ‑ test_metadata
src.tests.trainers.test_trainers.TestTrainSetFitClassifier ‑ test_multi_target
src.tests.trainers.test_trainers.TestTrainSetFitClassifier ‑ test_peft
src.tests.trainers.test_trainers.TestTrainer ‑ test_empty_name
src.tests.trainers.test_trainers.TestTrainer ‑ test_logging[False-17]
src.tests.trainers.test_trainers.TestTrainer ‑ test_logging[True-22]
src.tests.trainers.test_trainers.TestTrainer ‑ test_train_on_iterable_column
src.tests.trainers.test_trainers.TestTrainer ‑ test_trainer_not_in_datadreamer
src.tests.utils.test_device_utils.TestDeviceUtils ‑ test_device_id_to_true_device_id
src.tests.utils.test_device_utils.TestDeviceUtils ‑ test_device_to_device_id
src.tests.utils.test_device_utils.TestDeviceUtils ‑ test_get_device_env_variables
src.tests.utils.test_device_utils.TestDeviceUtils ‑ test_get_true_device_ids
src.tests.utils.test_device_utils.TestDeviceUtils ‑ test_is_cpu_device