Skip to content

Commit 2b71140

Browse files
committed
Reformat using ruff
1 parent 737e280 commit 2b71140

File tree

2 files changed

+16
-17
lines changed

2 files changed

+16
-17
lines changed

datacls/__init__.py

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -89,18 +89,17 @@ class Overloads:
8989
)
9090
"""
9191

92+
import dataclasses as dc
9293
from dataclasses import (
9394
MISSING,
9495
asdict,
9596
astuple,
9697
dataclass,
97-
field as _field,
9898
fields,
99-
make_dataclass as _make_dataclass,
10099
replace,
101100
)
102-
import functools
103-
from functools import cached_property
101+
from functools import cached_property, partial, wraps
102+
104103
import xmod
105104

106105
__all__ = (
@@ -125,24 +124,24 @@ class Overloads:
125124
_DFLT_ERR = 'Just one of default, default_factory and default_value may be set'
126125

127126

128-
@functools.wraps(dataclass)
127+
@wraps(dataclass)
129128
def mutable(cls=None, **kwargs):
130129
if not cls:
131-
return functools.partial(mutable, **kwargs)
130+
return partial(mutable, **kwargs)
132131

133132
return add_methods(dataclass(cls, **kwargs))
134133

135134

136-
@functools.wraps(_field)
135+
@wraps(dc.field)
137136
def field(default_factory=MISSING, **kwargs):
138-
return _field(default_factory=default_factory, **kwargs)
137+
return dc.field(default_factory=default_factory, **kwargs)
139138

140139

141140
def add_methods(dcls):
142141
"""Adds dataclasses functions as methods to a dataclass.
143142
144-
Adds three new instance methods: `asdict()`, `astuple()`, `replace()`
145-
and one new class method, `fields()`
143+
Adds three new instance methods: `asdict()`, `astuple()`, `replace()`
144+
and one new class method, `fields()`
146145
"""
147146
methods = (m for m in _METHODS if not hasattr(dcls, m))
148147
for m in methods:
@@ -153,17 +152,16 @@ def add_methods(dcls):
153152
return dcls
154153

155154

156-
@functools.wraps(_make_dataclass)
155+
@wraps(dc.make_dataclass)
157156
def make_dataclass(*a, **ka):
158-
return add_methods(_make_dataclass(*a, **ka))
157+
return add_methods(dc.make_dataclass(*a, **ka))
159158

160159

161-
hidden = functools.partial(field, compare=False, init=False, repr=False)
162-
immutable = functools.partial(mutable, frozen=True)
160+
hidden = partial(field, compare=False, init=False, repr=False)
161+
immutable = partial(mutable, frozen=True)
163162
dtyper = None
164163

165164

166-
167165
class Datacls:
168166
__call__ = staticmethod(immutable)
169167

test_datacls.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
1-
import datacls
21
from typing import Dict
32

3+
import datacls
4+
45

56
@datacls
67
class One:
@@ -124,7 +125,7 @@ def test_make_dataclass():
124125
C = datacls.make_dataclass(
125126
'C',
126127
[('x', int), 'y', ('z', int, datacls.field(default=5))],
127-
namespace={'add_one': lambda self: self.x + 1}
128+
namespace={'add_one': lambda self: self.x + 1},
128129
)
129130

130131
assert sorted(c.name for c in C.fields()) == ['x', 'y', 'z']

0 commit comments

Comments
 (0)