🐛 [BUG] Error: 'Key 'dtype_choices' is not in struct'

[theworld3]

Description

I wanted to excute nnsmith.model_gen mgen.max_nodes=5 mgen.dtype_choices="[f32]" model.type=onnx debug.viz=false mgen.save=$path. But it reurned :

Could not override 'mgen.dtype_choices'.
To append to your config use +mgen.dtype_choices=[f32]
Key 'dtype_choices' is not in struct
    full_key: mgen.dtype_choices
    object_type=dict

Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

When I excuted without mgen.dtype_choices="[f32]", it can work correctly. Could anyone tell me what should I do?

Installation

pip install "nnsmith[torch,onnx]" --upgrade

Reproduction

# Paste the commands or python script for reproducing the issue.

Screenshots

![DESCRIPTION](LINK.png)

Logs

No response

Front-end framework

No response

Version of the front-end framework

No response

Back-end engine

No response

Version of the front-end engine

No response

Other context

No response

[ganler]

Hi @XCY-Joestar it seems you directly installed nnsmith via PyPI. Maybe you can temporarily install nnsmith from GitHub HEAD to use this feature:

pip install "git+https://github.com/ise-uiuc/nnsmith@main#egg=nnsmith[torch,onnx]" --upgrade
# [optional] add more front- and back-ends such as [tf] and [tvm,ort,xla,...] in "[...]"

My apologies for not uptating the PyPI channel timely. We will recently release the v0.1.0 version where such features are enabled by default.

[theworld3]

@ganler Thank you for your reply. I can use nnsmtih correctly after pip install "git+https://github.com/ise-uiuc/nnsmith@main#egg=nnsmith[torch,onnx]" --upgrade