Does not work with generator functions? about memory_profiler

[fabianp]

yes that's a bug. Thanks for filling the issue.

[jonasc]

I still see that it does not work for generator functions (version 0.41 from pip). Can it be fixed?

[fabianp]

It should work, there are even some tests for this case. What is the error / failure?

[jonasc]

Consider the following code:

from memory_profiler import profile

@profile
def gen():
    yield 1
    yield 2
    return 3

@profile
def stuff():
    x = yield from gen()
    print('>', x)

print(list(stuff()))

Running it gives the following output (without any profiling output)

$ python -m memory_profiler test.py
> 3
[1, 2]

[fabianp]

We don't support the new "yield from" syntax unfortunately. Patches welcome
:-)

(BTW don't import profile if you run with "-m memory_profiler", it can have
bad side effects)

On Tue, Jul 19, 2016 at 3:20 PM, jonasc [email protected] wrote:

Consider the following code:

from memory_profiler import profile
@profiledef gen():
yield 1
yield 2
return 3
@profiledef stuff():
x = yield from gen()
print('>', x)
print(list(stuff()))

Running it gives the following output (without any profiling output)

$ python -m memory_profiler test.py

3
[1, 2]

—
You are receiving this because you commented.
Reply to this email directly, view it on GitHub
https://github.com/fabianp/memory_profiler/issues/42#issuecomment-233629778,
or mute the thread
https://github.com/notifications/unsubscribe-auth/AAQ8h5EkTaKuk-KmyE16QxPwcaV94LGPks5qXM8WgaJpZM4AjmU_
.

[astrojuanlu]

Change the title to "does not work with coroutines"? :)

[yarikoptic]

what am I dong wrong?

$> cat script2.py
import memory_profiler as mp
print(mp.__version__)
from memory_profiler import profile

@profile
def my_func():
    a = [1] * (10 ** 6)
    b = [2] * (2 * 10 ** 7)
    yield b
    del b
    yield a

for f in my_func():
   print('.')

$> python -m memory_profiler script2.py
0.52.0
.
.

[wowkin2]

Any progress on that?

[delijati]

#255

[Templarrr]

Still see this problem with generators (not coroutines) both in py2 and py3 in my code. profile wrapper just outputs nothing

E.g. script

from memory_profiler import profile

@profile
def gen():
    for i in range(100):
        yield i

@profile
def func1():
    return [i for i in gen()]

@profile
def func2():
    return list(range(100))

assert func1() == func2()

outputs for func2 and func1but nothing for gen

[luizoti]

The problem persistis, python 3.11.