Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 5 additions & 3 deletions src/squlearn/encoding_circuit/encoding_circuit_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,14 +149,16 @@ def draw(
):
feature_vec = ParameterVector(feature_label, self.num_encoding_slots)

elif self.num_features is not None and num_features is None:
feature_vec = ParameterVector(feature_label, self.num_features)
elif num_features or self.num_features:
feature_vec = ParameterVector(feature_label, num_features or self.num_features)
else:
feature_vec = [Parameter(feature_label)]

# ensure random configuration is available
if hasattr(self, "_is_config_available") and not self._is_config_available:
self._gen_random_config(num_features=num_features, seed=self.get_params()["seed"])
self._gen_random_config(
num_features=num_features or self.num_features or 0, seed=self.get_params()["seed"]
)

# ensure that the LayeredEncodingCircuit is built before drawing
if hasattr(self, "_build_layered_pqc"):
Expand Down
Loading