Skip to content

Commit

Permalink
revert embeddings changes
Browse files Browse the repository at this point in the history
  • Loading branch information
nmheim committed Nov 3, 2023
1 parent efcf349 commit c2abbd2
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions qadence/blocks/embedding.py
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,6 @@ def embedding_fn(params: StrTensorDict, inputs: StrTensorDict) -> StrTensorDict:
for e in constant_expressions + unique_const_matrices:
embedded_params[e] = params[stringify(e)]

embedded_params = {k: v for k, v in embedded_params.items()}
if to_gate_params:
gate_lvl_params: StrTensorDict = {}
for uuid, e in uuid_to_expr.items():
Expand All @@ -134,4 +133,4 @@ def embedding_fn(params: StrTensorDict, inputs: StrTensorDict) -> StrTensorDict:
for expr in unique_const_matrices
}
)
return {k: v for k, v in params.items()}, embedding_fn
return params, embedding_fn

0 comments on commit c2abbd2

Please sign in to comment.