README.md
pyproject.toml
src/lhcb_rex/__init__.py
src/lhcb_rex/py.typed
src/lhcb_rex.egg-info/PKG-INFO
src/lhcb_rex.egg-info/SOURCES.txt
src/lhcb_rex.egg-info/dependency_links.txt
src/lhcb_rex.egg-info/requires.txt
src/lhcb_rex.egg-info/top_level.txt
src/lhcb_rex/data/data_handler.py
src/lhcb_rex/data/data_handlerGENERALISED.py
src/lhcb_rex/data/data_manager.py
src/lhcb_rex/data/graph_maker.py
src/lhcb_rex/data/loader.py
src/lhcb_rex/data/batch/make_graphs_batch.py
src/lhcb_rex/data/batch/submit_tools.py
src/lhcb_rex/inference/inference_graph_constructor.py
src/lhcb_rex/inference/inference_network_manager.py
src/lhcb_rex/inference/params.py
src/lhcb_rex/inference/run_network.py
src/lhcb_rex/inference/runner.py
src/lhcb_rex/models/hetero_graph_based_GAN.py
src/lhcb_rex/models/hetero_graph_based_GAN_smearing.py
src/lhcb_rex/models/hetero_graph_based_VAE_smearing.py
src/lhcb_rex/models/hetero_graph_based_diffusion_model.py
src/lhcb_rex/models/hetero_graph_based_diffusion_model_simpler.py
src/lhcb_rex/models/hetero_graph_based_diffusion_model_smearing.py
src/lhcb_rex/models/model_handler.py
src/lhcb_rex/models/vanilla_GAN.py
src/lhcb_rex/models/vanilla_diffusion.py
src/lhcb_rex/processing/conditional_vars.py
src/lhcb_rex/processing/min_maxes.pkl
src/lhcb_rex/processing/processing_tools.py
src/lhcb_rex/processing/transfomer_quantiles.pkl
src/lhcb_rex/processing/transformers.py
src/lhcb_rex/rapidsim/rapidsim_tools.py
src/lhcb_rex/rapidsim/run_rapidsim.py
src/lhcb_rex/settings/PVsmear_model_parameters.json
src/lhcb_rex/settings/default_model_parameters.json
src/lhcb_rex/settings/diffusion_model_parameters.json
src/lhcb_rex/settings/diffusion_model_parameters_vtx.json
src/lhcb_rex/settings/globals.py
src/lhcb_rex/settings/smear_diffusion_model_parameters.json
src/lhcb_rex/settings/smearing_model_parameters.json
src/lhcb_rex/tests/test_basic.py
src/lhcb_rex/tools/display.py
src/lhcb_rex/tools/get_weights.py
src/lhcb_rex/tools/stopwatch.py
src/lhcb_rex/tools/tuple_tools.py
src/lhcb_rex/tools/variables_tools.py
src/lhcb_rex/training/dataloader_handler.py
src/lhcb_rex/training/hetero_trainer.py
src/lhcb_rex/training/train.py
src/lhcb_rex/training/trainer.py