Skip to main content

pyo3 Troubles with `inspect.signature` method

I have a rust struct with which I had implemented python's magic methods __iter__, __next__, and __call__.

#[pyclass(name = "PyNRICContainer")]
#[derive(Debug)]
#[pyo3(text_signature = "(cls, value, values)")]
pub struct PyNRICContainer {
    #[pyo3(get, set)]
    boolean: bool,
}

Here are the implementations:

#[pymethods]
impl PyNRICContainer {

    #[new]
    #[pyo3(signature = "(cls, value, values)")]
    fn new() -> PyResult<PyNRICContainer> {
        Ok(PyNRICContainer { boolean: true })
    }

    fn __iter__(slf: PyRef<'_, Self>) -> PyRef<'_, Self> {
        slf
    }

    fn __next__(mut slf: PyRefMut<'_, Self>) -> IterNextOutput<PyRefMut<'_, Self>, &'static str> {
        if slf.boolean {
            slf.boolean = false;
            IterNextOutput::Yield(slf)
        } else {
            IterNextOutput::Return("No Longer Iterable.")
        }
    }


    #[classmethod]
    fn __call__(&mut self, py: Python<'_>, value: &PyString, values: Option<&PyDict>) -> PyResult<PyNRIC> {
        let v: String = value.extract()?;
        PyNRIC::new(v)
    }
    
}

With this, I enabled the behavior of yield this object through the following python codes.

g = PyNRICContainer()
gs = (
      *(g if g else ()),
  )
  print("gs", gs)

The output of the above snippet is:

gs (<builtins.PyNRICContainer object at 0x0000021250B37650>,)

Now, as inspect.signature method cannot get a signature out of g in the above snippet, it raises a ValueError shown below.

ValueError: callable <builtins.PyNRICContainer object at 0x0000021250B37650> is not supported by signature

In the pyo3's documentations, there is a mention of the macro #[pyo3(text_signature)], I tried to implement it above the fn __call__ but it couldn't compile as:

error: `text_signature` cannot be used with `__call__`

My objective is to have a python callable object which is also 'yieldable', i.e. I should be able to yield the object, e.g. yield g in a function or using to * operator to unpack the function from which g was yielded.

E.g.

def yield_g():
    yield g

gs = (
    *(yield_g() if yield_g else None),
)

And it should output something like (g,). I have also tried printing type(g).text_signature but it gives an AttributeError.

Is it possible to make a pyo3::types::PyFunction a PyIterator such that it is yieldable?



source https://stackoverflow.com/questions/75574260/pyo3-troubles-with-inspect-signature-method

Comments

Popular posts from this blog

ValueError: X has 10 features, but LinearRegression is expecting 1 features as input

So, I am trying to predict the model but its throwing error like it has 10 features but it expacts only 1. So I am confused can anyone help me with it? more importantly its not working for me when my friend runs it. It works perfectly fine dose anyone know the reason about it? cv = KFold(n_splits = 10) all_loss = [] for i in range(9): # 1st for loop over polynomial orders poly_order = i X_train = make_polynomial(x, poly_order) loss_at_order = [] # initiate a set to collect loss for CV for train_index, test_index in cv.split(X_train): print('TRAIN:', train_index, 'TEST:', test_index) X_train_cv, X_test_cv = X_train[train_index], X_test[test_index] t_train_cv, t_test_cv = t[train_index], t[test_index] reg.fit(X_train_cv, t_train_cv) loss_at_order.append(np.mean((t_test_cv - reg.predict(X_test_cv))**2)) # collect loss at fold all_loss.append(np.mean(loss_at_order)) # collect loss at order plt.plot(np.log(al...

Sorting large arrays of big numeric stings

I was solving bigSorting() problem from hackerrank: Consider an array of numeric strings where each string is a positive number with anywhere from to digits. Sort the array's elements in non-decreasing, or ascending order of their integer values and return the sorted array. I know it works as follows: def bigSorting(unsorted): return sorted(unsorted, key=int) But I didnt guess this approach earlier. Initially I tried below: def bigSorting(unsorted): int_unsorted = [int(i) for i in unsorted] int_sorted = sorted(int_unsorted) return [str(i) for i in int_sorted] However, for some of the test cases, it was showing time limit exceeded. Why is it so? PS: I dont know exactly what those test cases were as hacker rank does not reveal all test cases. source https://stackoverflow.com/questions/73007397/sorting-large-arrays-of-big-numeric-stings

How to load Javascript with imported modules?

I am trying to import modules from tensorflowjs, and below is my code. test.html <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Document</title </head> <body> <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.0.0/dist/tf.min.js"></script> <script type="module" src="./test.js"></script> </body> </html> test.js import * as tf from "./node_modules/@tensorflow/tfjs"; import {loadGraphModel} from "./node_modules/@tensorflow/tfjs-converter"; const MODEL_URL = './model.json'; const model = await loadGraphModel(MODEL_URL); const cat = document.getElementById('cat'); model.execute(tf.browser.fromPixels(cat)); Besides, I run the server using python -m http.server in my command prompt(Windows 10), and this is the error prompt in the console log of my browser: Failed to loa...