make_dataclass for Pytree

[pablo2909]

Hi,

Thank you for this super cool library.

I was wondering if there is something similar to make_dataclass from dataclasses here ? In my case it would be super useful to build some custom metrics based on a config file.

Thanks again :)

[pablo2909]

Playing a bit with it, I'm able to do this but I'm not sure this is the way to go:

import dataclasses as dc
from simple_pytree import dataclass, Pytree, static_field

A = dc.make_dataclass('A',
                   [('x', int),
                     'y',
                    ('z', int, static_field(default=5))],
                   )

@dataclass
class B(A, Pytree):
    ...

B(1, 2, 3)

[cgarciae]

Hey @pablo2909, I think your approach is correct and should be supported. It currently doesn't work but I created #14 fix it.

[cgarciae]

Fixed in 0.2.2

[pablo2909]

Thank you for the super quick response and fix :)

Could you please elaborate on what did not work and the fix you brought ? Because the MWE I posted was working for me.

Thanks again ! This library is really super useful :)

[cgarciae]

No problem! The resulting pytree was not makingz a static field.