search

aai-institute / nnbench

A small framework for benchmarking machine learning models.
https://aai-institute.github.io/nnbench/
Apache License 2.0
10 stars 3 forks source link

(fix) Change default == value to is #139

Closed maxmynter closed 6 months ago

maxmynter commented 6 months ago

as the default can be an array. If that is the case the previous implementation fails in the if clause as the truth value of an array is ambiguous.

Previously the following would fail: 

@nnbench.parametrize(({"a": np.zeros((10**5,))},))
def matsquare(a):
    return a @ a