From a8457f589e243c03afb015b12523dcfd55226c9e Mon Sep 17 00:00:00 2001 From: Isuru Haththotuwa Date: Mon, 28 Oct 2024 09:06:10 +0100 Subject: [PATCH] enable gaussian input in image_data function --- cirkit/templates/circuit_templates/data.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cirkit/templates/circuit_templates/data.py b/cirkit/templates/circuit_templates/data.py index 6e026f93..4c29d8b5 100644 --- a/cirkit/templates/circuit_templates/data.py +++ b/cirkit/templates/circuit_templates/data.py @@ -64,7 +64,7 @@ def image_data( "poon-domingos", ]: raise ValueError(f"Unknown region graph called {region_graph}") - if input_layer not in ["categorical", "binomial", "embedding"]: + if input_layer not in ["categorical", "binomial", "embedding", "gaussian"]: raise ValueError(f"Unknown input layer called {input_layer}") # Construct the image-tailored region graph @@ -79,6 +79,8 @@ def image_data( input_kwargs = {"total_count": 255} case "embedding": input_kwargs = {"num_states": 256} + case "gaussian": + input_kwargs = {} case _: assert False if input_params is not None: