I would like to work on this issue. Thanks.

I would like to work on this issue. Thanks.

Feel free to open a PR

How do I test for the jax and pytorch backend scripts? Cause when I run the test suite in the test directory I don't see those scripts being run. Thanks.

How do I test for the jax and pytorch backend scripts? Cause when I run the test suite in the test directory I don't see those scripts being run. Thanks.

You need to install those packages as they're optional. Otherwise the tests are skipped automatically

I’m currently working on the issue and have noticed a few things. While the FunctionGraph’s handling of inputs and outputs may seem unnecessary, it actually improves code writability. When initializing the FunctionGraph, we can simply provide the outputs to the graph and the corresponding inputs are understood by the graph implicitly. This is especially helpful when dealing with many inputs, saving time and effort. IMHO Unless removing the FunctionGraph significantly enhances performance or memory usage of the test, I believe we should keep it. Thank you.

You need to specify test values for the inputs so there's already a need to manually specify the inputs anyway.

If we are removing the set_test_value machinery then that would mean all the inputs we are passing would be constants right?(not variables)
Or is there a better way to initialise variables without using set_test_value machinery?
And if we are just passing constants as inputs into the output then we would not need to pass input_type and inputs in to the modified compare_py_and_x function right?
Thanks.

The variables can be initialized the very same way without the test values attached to it. Don't need (and shouldn't) be replaced by constants

compare_numba_and_py( inputs, [g], [ i.tag.test_value for i in inputs if not isinstance(i, SharedVariable | Constant) ], )

@ricardoV94 In a lot of places the input values are set through the test values of variables (especially when we use @pytest.mark.parametrize), if the test values are not attached to the variables how should we access them?
Thanks.

Change parametrize to be a tuple of (variable, test_values) instead

@ricardoV94 please let me know about further steps to be taken. Thanks.