Is there any way to return a function that will execute on the python side when returned?

[varchasgopalaswamy]

I have a feeling this would not be possible, but I wanted to check.

Lets say I have a python function

def add_units_to_array(array, units:str): 
    return array.with_units(units)

and I have a Rust struct

struct MyObject { 
     array: Array3<f64>
     units: String
}
impl MyObject { 
      
     #[getter] 
     fn array_with_units(&self) {
     /// What goes here? 
}
}

I would like to be able to do

obj = Object(np.linspace(0,1,100), "cm") 
obj.array_with_units # returns add_units_to_array(obj.array, obj.units)

I assume array would need to be cloned, which I would be fine with. I would also settle for being able to achieve this with

obj.array_with_units()

[davidhewitt]

Looks like you need to find a way to get a reference to add_units_to_array (as f) and then use f.call1((self.array, self.units)). You could either pass that to the Object constructor or import it from wherever the Python code is.

[varchasgopalaswamy]

Thanks @davidhewitt! The snippet below worked perfectly. This package is really something!

    #[getter]
    fn py_simulation_time_with_units<'py>(&self, py: Python<'py>) -> Result<Py<PyAny>, PyErr> {
        let units = "s";
        let py_array = self.simulation_time.to_pyarray(py);
        Python::attach(|py| {
            let fun: Py<PyAny> = PyModule::from_code(
                py,
                c_str!(
                    "from package import unit_registry as ureg
def array_to_units(array, units):
    return array * ureg(units)
                "
                ),
                c_str!("array_to_units.py"),
                c_str!(""),
            )?
            .getattr("array_to_units")?
            .into();
            return fun.call1(py, (py_array, units));
        })
    }