Skip to content

Commit 12546c7

Browse files
committed
Do not test json serialisation of *Fit features
1 parent 64ecde4 commit 12546c7

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

light-curve/tests/light_curve_ext/test_feature.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -144,13 +144,16 @@ def mutation(value):
144144
return objects
145145

146146

147-
def gen_feature_evaluators(*, parametric_variants=0, rng=None):
147+
def gen_feature_evaluators(*, parametric_variants=0, skip_fit=False, rng=None):
148148
if parametric_variants == 0:
149149
for cls in non_param_feature_classes:
150150
yield cls()
151151
return
152152
rng = np.random.default_rng(rng)
153-
for cls in all_feature_classes:
153+
classes = all_feature_classes
154+
if skip_fit:
155+
classes = classes - fit_feature_classes
156+
for cls in classes:
154157
yield from construct_example_objects(cls, parametric_variants=parametric_variants, rng=rng)
155158

156159

0 commit comments

Comments
 (0)