LICENSE
README.md
pyproject.toml
piano/__init__.py
piano/__main__.py
piano/models/__init__.py
piano/models/base_models.py
piano/models/base_modules.py
piano/models/scvi_models.py
piano/utils/__init__.py
piano/utils/composer.py
piano/utils/covariates.py
piano/utils/data.py
piano/utils/io.py
piano/utils/knn.py
piano/utils/preprocessing.py
piano/utils/scvi_mode.py
piano/utils/timer.py
piano/utils/triton_sparse.py
piano/utils/umap.py
piano_integration.egg-info/PKG-INFO
piano_integration.egg-info/SOURCES.txt
piano_integration.egg-info/dependency_links.txt
piano_integration.egg-info/requires.txt
piano_integration.egg-info/top_level.txt
tests/test_backed_mode.py
tests/test_categorical_encoding.py
tests/test_composer_arguments.py
tests/test_concat_dataset_index.py
tests/test_count_storage.py
tests/test_counterfactual_covariates.py
tests/test_counterfactual_label_types.py
tests/test_decoder_dropout.py
tests/test_dependencies.py
tests/test_determinism.py
tests/test_duplicate_csr_entries.py
tests/test_empty_adatas.py
tests/test_epoch_losses.py
tests/test_gene_selection.py
tests/test_highly_variable_genes.py
tests/test_input_formats.py
tests/test_invalid_categorical_covariates.py
tests/test_invalid_continuous_covariates.py
tests/test_invalid_counts.py
tests/test_invalid_sparse_continuous_covariates.py
tests/test_io.py
tests/test_latent_representation.py
tests/test_lazy_imports.py
tests/test_matmul_precision.py
tests/test_memory_mode_fallback.py
tests/test_missing_categorical_values.py
tests/test_model_not_ready.py
tests/test_no_adversarial_training.py
tests/test_no_best_model.py
tests/test_no_covariates.py
tests/test_num_workers.py
tests/test_prepared_model_device.py
tests/test_query_checks.py
tests/test_save_load.py
tests/test_scvi_mode.py
tests/test_scvi_model.py
tests/test_several_training_adatas.py
tests/test_sparse_cuda_matrix.py
tests/test_stratified_sampling.py
tests/test_training_checkpoints.py
tests/test_training_defaults.py
tests/test_training_schedule.py
tests/test_umap.py
tests/test_unlabeled_value.py
tests/test_unpickling_allowlist.py
tests/test_unseen_sparse_categories.py
tests/test_untrained_model_warning.py
tests/test_zinb_counterfactuals.py
tests/test_zinb_log_prob.py