LICENSE
README.md
pyproject.toml
hydra_plugins/teia_searchpath.py
pyteia.egg-info/PKG-INFO
pyteia.egg-info/SOURCES.txt
pyteia.egg-info/dependency_links.txt
pyteia.egg-info/entry_points.txt
pyteia.egg-info/requires.txt
pyteia.egg-info/top_level.txt
src/teia/__main__.py
src/teia/callbacks.py
src/teia/base/__init__.py
src/teia/base/deps.py
src/teia/base/envelope.py
src/teia/base/fields.py
src/teia/base/keys.py
src/teia/base/protocols.py
src/teia/base/schema.py
src/teia/base/task.py
src/teia/base/data/__init__.py
src/teia/base/data/cache.py
src/teia/base/data/nodes.py
src/teia/base/eval/__init__.py
src/teia/base/eval/nodes.py
src/teia/base/interactive/__init__.py
src/teia/base/interactive/batch.py
src/teia/base/interactive/env.py
src/teia/base/interactive/policy.py
src/teia/base/interactive/spaces.py
src/teia/base/net/__init__.py
src/teia/base/net/action.py
src/teia/base/net/node.py
src/teia/conf/datamodule/tabular-cls/csv.yaml
src/teia/conf/datamodule/vision-cls/image_folder.yaml
src/teia/conf/evalmodule/_shared/decision_rule.yaml
src/teia/conf/evalmodule/_shared/run.yaml
src/teia/conf/evalmodule/audio-cls/default.yaml
src/teia/conf/evalmodule/audio-multi-cls/default.yaml
src/teia/conf/evalmodule/audio-wave-cls/default.yaml
src/teia/conf/evalmodule/graph-node-cls/default.yaml
src/teia/conf/evalmodule/tabular-cls/default.yaml
src/teia/conf/evalmodule/tabular-forecast/default.yaml
src/teia/conf/evalmodule/tabular-multi-cls/default.yaml
src/teia/conf/evalmodule/tabular-reg/default.yaml
src/teia/conf/evalmodule/tabular-survival/default.yaml
src/teia/conf/evalmodule/vision-cls/default.yaml
src/teia/conf/evalmodule/vision-det/default.yaml
src/teia/conf/evalmodule/vision-gen/default.yaml
src/teia/conf/evalmodule/vision-inst-seg/default.yaml
src/teia/conf/evalmodule/vision-multi-cls/default.yaml
src/teia/conf/evalmodule/vision-obb/default.yaml
src/teia/conf/evalmodule/vision-pose/default.yaml
src/teia/conf/evalmodule/vision-sem-seg/default.yaml
src/teia/conf/netmodule/tabular-cls/numerical_mlp.yaml
src/teia/conf/netmodule/vision-cls/mlp.yaml
src/teia/conf/netmodule/vision-cls/wide_mlp.yaml
src/teia/conf/node/data/augment/clamp_boxes.yaml
src/teia/conf/node/data/augment/mosaic.yaml
src/teia/conf/node/data/augment/random_h_flip.yaml
src/teia/conf/node/data/augment/random_h_flip_pose.yaml
src/teia/conf/node/data/augment/random_perspective_crop.yaml
src/teia/conf/node/data/augment/random_perspective_crop_obb.yaml
src/teia/conf/node/data/augment/sanitize_boxes.yaml
src/teia/conf/node/data/augment/spec_augment.yaml
src/teia/conf/node/data/collate/batch_index.yaml
src/teia/conf/node/data/collate/class_labels.yaml
src/teia/conf/node/data/collate/concat_boxes.yaml
src/teia/conf/node/data/collate/concat_boxes_5.yaml
src/teia/conf/node/data/collate/concat_classes.yaml
src/teia/conf/node/data/collate/concat_keypoints.yaml
src/teia/conf/node/data/collate/list_field.yaml
src/teia/conf/node/data/collate/multi_hot_classes.yaml
src/teia/conf/node/data/collate/overlap_masks.yaml
src/teia/conf/node/data/collate/sem_masks.yaml
src/teia/conf/node/data/collate/semantic_masks.yaml
src/teia/conf/node/data/collate/stack.yaml
src/teia/conf/node/data/collate/stack_float32.yaml
src/teia/conf/node/data/collate/stack_images.yaml
src/teia/conf/node/data/collate/stack_int64.yaml
src/teia/conf/node/data/join/key_join.yaml
src/teia/conf/node/data/reader/audio_folder_reader.yaml
src/teia/conf/node/data/reader/audio_folder_reader_multi_label.yaml
src/teia/conf/node/data/reader/coco.yaml
src/teia/conf/node/data/reader/csv_rows.yaml
src/teia/conf/node/data/reader/files_reader.yaml
src/teia/conf/node/data/reader/files_reader_json.yaml
src/teia/conf/node/data/reader/files_reader_masks.yaml
src/teia/conf/node/data/reader/files_reader_txt_labels.yaml
src/teia/conf/node/data/reader/image_folder_reader.yaml
src/teia/conf/node/data/reader/image_folder_reader_multi_label.yaml
src/teia/conf/node/data/reader/image_folder_reader_unlabeled.yaml
src/teia/conf/node/data/reader/labelme.yaml
src/teia/conf/node/data/transform/boolean.yaml
src/teia/conf/node/data/transform/categorical.yaml
src/teia/conf/node/data/transform/count.yaml
src/teia/conf/node/data/transform/decode_audio.yaml
src/teia/conf/node/data/transform/decode_image.yaml
src/teia/conf/node/data/transform/decode_image_with_ori_shape.yaml
src/teia/conf/node/data/transform/decode_mask.yaml
src/teia/conf/node/data/transform/encode_labels.yaml
src/teia/conf/node/data/transform/letterbox.yaml
src/teia/conf/node/data/transform/letterbox_pose.yaml
src/teia/conf/node/data/transform/letterbox_seg.yaml
src/teia/conf/node/data/transform/mel_spectrogram.yaml
src/teia/conf/node/data/transform/multi_hot_labels.yaml
src/teia/conf/node/data/transform/name_list_encode.yaml
src/teia/conf/node/data/transform/normalize_image.yaml
src/teia/conf/node/data/transform/normalize_pixel_norm.yaml
src/teia/conf/node/data/transform/numerical.yaml
src/teia/conf/node/data/transform/numerical_mask.yaml
src/teia/conf/node/data/transform/obb_from_polys.yaml
src/teia/conf/node/data/transform/parse_labelme.yaml
src/teia/conf/node/data/transform/parse_yolo_det.yaml
src/teia/conf/node/data/transform/parse_yolo_pose.yaml
src/teia/conf/node/data/transform/parse_yolo_seg.yaml
src/teia/conf/node/data/transform/rasterize_instance_masks.yaml
src/teia/conf/node/data/transform/resize_image.yaml
src/teia/conf/node/data/transform/resize_mask.yaml
src/teia/conf/node/data/transform/temporal.yaml
src/teia/conf/node/data/transform/temporal_mask.yaml
src/teia/conf/node/data/transform/yolo_meta.yaml
src/teia/conf/node/data/writer/csv_writer.yaml
src/teia/conf/node/eval/kernel/iou_mask.yaml
src/teia/conf/node/eval/kernel/iou_rotated.yaml
src/teia/conf/node/eval/kernel/iou_xyxy.yaml
src/teia/conf/node/eval/kernel/oks.yaml
src/teia/conf/node/eval/metric/calibration.yaml
src/teia/conf/node/eval/metric/concordance.yaml
src/teia/conf/node/eval/metric/confusion.yaml
src/teia/conf/node/eval/metric/distribution_test.yaml
src/teia/conf/node/eval/metric/frechet_distance.yaml
src/teia/conf/node/eval/metric/gen_diagnostics.yaml
src/teia/conf/node/eval/metric/horizon_error.yaml
src/teia/conf/node/eval/metric/instance_match.yaml
src/teia/conf/node/eval/metric/kernel_distance.yaml
src/teia/conf/node/eval/metric/km_estimate.yaml
src/teia/conf/node/eval/metric/ranked_average_precision.yaml
src/teia/conf/node/eval/metric/residuals.yaml
src/teia/conf/node/eval/metric/semantic_confusion.yaml
src/teia/conf/node/eval/metric/target_summary.yaml
src/teia/conf/node/eval/metric/threshold_optimization.yaml
src/teia/conf/node/eval/metric/training_curves.yaml
src/teia/conf/node/eval/view/binary_confusion_grid.yaml
src/teia/conf/node/eval/view/calibration_diagram.yaml
src/teia/conf/node/eval/view/grouped_histogram.yaml
src/teia/conf/node/eval/view/matrix_heatmap.yaml
src/teia/conf/node/eval/view/multi_curve_bundle.yaml
src/teia/conf/node/eval/view/ranked_bar.yaml
src/teia/conf/node/eval/view/scatter_plot.yaml
src/teia/conf/node/eval/view/table.yaml
src/teia/conf/node/eval/view/threshold_curve.yaml
src/teia/conf/node/net/act/categorical.yaml
src/teia/conf/node/net/act/gaussian.yaml
src/teia/conf/node/net/act/seg_softmax.yaml
src/teia/conf/node/net/act/sigmoid.yaml
src/teia/conf/node/net/act/softmax.yaml
src/teia/conf/node/net/encoder/mlp_image.yaml
src/teia/conf/node/net/encoder/mlp_tabular.yaml
src/teia/conf/node/net/fusion/concat_flatten.yaml
src/teia/conf/node/net/fusion/concat_last.yaml
src/teia/conf/node/net/head/linear.yaml
src/teia/conf/node/net/head/linear_cls.yaml
src/teia/conf/node/net/head/pointwise_mlp.yaml
src/teia/conf/node/net/loss/bce.yaml
src/teia/conf/node/net/loss/cross_entropy.yaml
src/teia/conf/node/net/loss/gaussian_nll.yaml
src/teia/conf/node/net/loss/mask_cross_entropy.yaml
src/teia/conf/node/net/stem/flatten.yaml
src/teia/conf/optim/adam_1e3.yaml
src/teia/conf/task/tabular-cls.yaml
src/teia/conf/task/vision-cls.yaml
src/teia/core/__init__.py
src/teia/core/callbacks.py
src/teia/core/cli.py
src/teia/core/deps.py
src/teia/core/instantiate.py
src/teia/core/plugin.py
src/teia/core/torch_compat.py
src/teia/core/utils.py
src/teia/core/capture/__init__.py
src/teia/core/capture/callback.py
src/teia/core/capture/capture_io.py
src/teia/core/capture/extract.py
src/teia/core/capture/rasters.py
src/teia/core/capture/serialize.py
src/teia/core/capture/store.py
src/teia/core/capture/walltime.py
src/teia/core/commands/__init__.py
src/teia/core/commands/_runtime.py
src/teia/core/commands/config.py
src/teia/core/commands/eval.py
src/teia/core/commands/export.py
src/teia/core/commands/infer.py
src/teia/core/commands/init.py
src/teia/core/commands/plugin.py
src/teia/core/commands/test.py
src/teia/core/commands/train.py
src/teia/core/conf/config.yaml
src/teia/core/conf/callbacks/default.yaml
src/teia/core/conf/compare/default.yaml
src/teia/core/conf/export/default.yaml
src/teia/core/conf/infer/default.yaml
src/teia/core/conf/loggers/default.yaml
src/teia/core/conf/runtime/default.yaml
src/teia/core/conf/test/default.yaml
src/teia/core/conf/train/default.yaml
src/teia/core/conf/trainer/default.yaml
src/teia/core/config/__init__.py
src/teia/core/config/explain.py
src/teia/core/config/layers.py
src/teia/core/config/lint.py
src/teia/core/config/scaffold.py
src/teia/core/config/schema.py
src/teia/core/datamodule/__init__.py
src/teia/core/datamodule/executor.py
src/teia/core/datamodule/graph.py
src/teia/core/datamodule/interactive/__init__.py
src/teia/core/datamodule/interactive/_collect.py
src/teia/core/datamodule/interactive/_tensor_ops.py
src/teia/core/datamodule/interactive/async_runner.py
src/teia/core/datamodule/interactive/env_reader.py
src/teia/core/datamodule/interactive/policy.py
src/teia/core/datamodule/interactive/regime.py
src/teia/core/datamodule/interactive/replay.py
src/teia/core/datamodule/interactive/rollout.py
src/teia/core/datamodule/interactive/runner_base.py
src/teia/core/datamodule/interactive/sync.py
src/teia/core/eval/__init__.py
src/teia/core/eval/compare_runner.py
src/teia/core/eval/graph.py
src/teia/core/eval/runner.py
src/teia/core/eval/streaming.py
src/teia/core/export/assemble.py
src/teia/core/export/bundle.py
src/teia/core/export/collect.py
src/teia/core/export/kernels.py
src/teia/core/export/onnx.py
src/teia/core/export/preprocess_slice.py
src/teia/core/infer/__init__.py
src/teia/core/infer/runner.py
src/teia/core/infer/writers.py
src/teia/core/interactive/__init__.py
src/teia/core/interactive/live_view.py
src/teia/core/interactive/online.py
src/teia/core/interactive/render.py
src/teia/core/module/__init__.py
src/teia/core/module/codegen.py
src/teia/core/module/dry_run.py
src/teia/core/module/net.py
src/teia/core/module/sync.py
src/teia/core/runtime/composer.py
src/teia/core/runtime/determinism.py
src/teia/core/runtime/engine.py
src/teia/core/runtime/hooks.py
src/teia/core/runtime/layout.py
src/teia/core/runtime/memory.py
src/teia/core/runtime/project_python.py
src/teia/core/runtime/run_descriptor.py
src/teia/core/runtime/validate.py
src/teia/core/templates/placeholders.py
src/teia/node/data/_frame.py
src/teia/node/data/_paths.py
src/teia/node/data/collate/geometric.py
src/teia/node/data/collate/tensor.py
src/teia/node/data/join/__init__.py
src/teia/node/data/join/align.py
src/teia/node/data/reader/audio.py
src/teia/node/data/reader/coco.py
src/teia/node/data/reader/csv.py
src/teia/node/data/reader/files.py
src/teia/node/data/reader/image_folder.py
src/teia/node/data/reader/labelme.py
src/teia/node/data/reader/parquet.py
src/teia/node/data/reader/rows.py
src/teia/node/data/reader/sql.py
src/teia/node/data/reader/extension/base.py
src/teia/node/data/reshape/window.py
src/teia/node/data/transform/decode.py
src/teia/node/data/transform/geometry.py
src/teia/node/data/transform/labels.py
src/teia/node/data/transform/normalize.py
src/teia/node/data/transform/obb.py
src/teia/node/data/transform/parse_labelme.py
src/teia/node/data/transform/parse_yolo.py
src/teia/node/data/transform/rasterize.py
src/teia/node/data/transform/schema.py
src/teia/node/data/transform/stft.py
src/teia/node/data/transform/yolo_meta.py
src/teia/node/data/transform/augment/__init__.py
src/teia/node/data/transform/augment/affine.py
src/teia/node/data/transform/augment/box_hygiene.py
src/teia/node/data/transform/augment/color.py
src/teia/node/data/transform/augment/flip.py
src/teia/node/data/transform/augment/mosaic.py
src/teia/node/data/transform/augment/spec_augment.py
src/teia/node/data/writer/__init__.py
src/teia/node/data/writer/class_dir.py
src/teia/node/data/writer/csv.py
src/teia/node/eval/compare/_shared.py
src/teia/node/eval/compare/curve_overlay.py
src/teia/node/eval/compare/factor_diff.py
src/teia/node/eval/compare/metric_table.py
src/teia/node/eval/compare/pareto.py
src/teia/node/eval/compare/seed_aggregate.py
src/teia/node/eval/kernel/_geometry.py
src/teia/node/eval/kernel/iou_mask.py
src/teia/node/eval/kernel/iou_rotated.py
src/teia/node/eval/kernel/iou_xyxy.py
src/teia/node/eval/kernel/oks.py
src/teia/node/eval/metric/aligned/_shared.py
src/teia/node/eval/metric/aligned/calibration.py
src/teia/node/eval/metric/aligned/concordance.py
src/teia/node/eval/metric/aligned/confusion.py
src/teia/node/eval/metric/aligned/horizon_error.py
src/teia/node/eval/metric/aligned/instance_match.py
src/teia/node/eval/metric/aligned/km_estimate.py
src/teia/node/eval/metric/aligned/ranked_average_precision.py
src/teia/node/eval/metric/aligned/residuals.py
src/teia/node/eval/metric/aligned/semantic_confusion.py
src/teia/node/eval/metric/aligned/threshold_optimization.py
src/teia/node/eval/metric/run/gen_diagnostics.py
src/teia/node/eval/metric/run/target_summary.py
src/teia/node/eval/metric/run/training_curves.py
src/teia/node/eval/metric/unpaired/_embedding.py
src/teia/node/eval/metric/unpaired/_shared.py
src/teia/node/eval/metric/unpaired/distribution_test.py
src/teia/node/eval/metric/unpaired/features.py
src/teia/node/eval/metric/unpaired/frechet_distance.py
src/teia/node/eval/metric/unpaired/kernel_distance.py
src/teia/node/eval/view/_shared.py
src/teia/node/eval/view/binary_confusion_grid.py
src/teia/node/eval/view/calibration_diagram.py
src/teia/node/eval/view/grouped_histogram.py
src/teia/node/eval/view/matrix_heatmap.py
src/teia/node/eval/view/multi_curve_bundle.py
src/teia/node/eval/view/ranked_bar.py
src/teia/node/eval/view/scatter_plot.py
src/teia/node/eval/view/summary.py
src/teia/node/eval/view/table.py
src/teia/node/eval/view/threshold_curve.py
src/teia/node/net/_activation_utils.py
src/teia/node/net/activation/categorical.py
src/teia/node/net/activation/gaussian.py
src/teia/node/net/activation/identity.py
src/teia/node/net/activation/segmentation.py
src/teia/node/net/activation/sigmoid.py
src/teia/node/net/activation/softmax.py
src/teia/node/net/loss/bce.py
src/teia/node/net/loss/cross_entropy.py
src/teia/node/net/loss/gaussian_nll.py
src/teia/node/net/loss/mae.py
src/teia/node/net/loss/mse.py
src/teia/node/net/loss/seg_cross_entropy.py
src/teia/node/net/many_to_one/_base.py
src/teia/node/net/many_to_one/agnostic/concat.py
src/teia/node/net/many_to_one/agnostic/gated.py
src/teia/node/net/many_to_one/agnostic/sum.py
src/teia/node/net/one_to_many/agnostic/split.py
src/teia/node/net/one_to_one/_base.py
src/teia/node/net/one_to_one/agnostic/flatten.py
src/teia/node/net/one_to_one/agnostic/linear.py
src/teia/node/net/one_to_one/agnostic/mlp.py
src/teia/node/net/one_to_one/agnostic/pointwise_mlp.py
src/teia/skills/teia/SKILL.md
src/teia/skills/teia/baselines.md
src/teia/skills/teia/inspect.md
src/teia/skills/teia/make-a-plugin.md
src/teia/skills/teia/publish.md
src/teia/skills/teia/run.md
src/teia/skills/teia/swap.md
src/teia/skills/teia/use-a-task.md
src/teia/skills/teia/write-component.md
src/teia/skills/teia/write-config.md
src/teia/task/_bases.py
src/teia/task/audio_cls.py
src/teia/task/audio_multi_cls.py
src/teia/task/audio_wave_cls.py
src/teia/task/graph_node_cls.py
src/teia/task/tabular_cls.py
src/teia/task/tabular_forecast.py
src/teia/task/tabular_multi_cls.py
src/teia/task/tabular_reg.py
src/teia/task/tabular_survival.py
src/teia/task/vision_cls.py
src/teia/task/vision_det.py
src/teia/task/vision_gen.py
src/teia/task/vision_inst_seg.py
src/teia/task/vision_multi_cls.py
src/teia/task/vision_obb.py
src/teia/task/vision_pose.py
src/teia/task/vision_sem_seg.py
tests/test_augment_nodes.py
tests/test_base_activation.py
tests/test_capture_map.py
tests/test_capture_store.py
tests/test_checkpoint_restore.py
tests/test_classification_thresholds.py
tests/test_config_commands.py
tests/test_config_lint.py
tests/test_config_tree.py
tests/test_core_callbacks.py
tests/test_data_graph.py
tests/test_eval_bundle_cls.py
tests/test_eval_bundle_smoke.py
tests/test_eval_compare.py
tests/test_eval_graph.py
tests/test_eval_instance_match.py
tests/test_eval_metrics_aligned.py
tests/test_eval_preflight_and_run_descriptor.py
tests/test_eval_semantic_confusion.py
tests/test_eval_unpaired_metrics.py
tests/test_eval_views.py
tests/test_export_bundle.py
tests/test_join.py
tests/test_lightning_logging.py
tests/test_plugin_check.py
tests/test_rollout_buffer.py
tests/test_runtime.py
tests/test_streaming_val_metrics.py
tests/test_task_contract.py
tests/test_teia_eval_package.py
tests/test_teia_init.py
tests/test_weight_averaging.py