Python test_export.py code style fix

forestry-labs · Aug 14, 2023 · 79525ba · 79525ba
1 parent 7f84382
commit 79525ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Python/tests/test_export.py b/Python/tests/test_export.py
@@ -17,7 +17,7 @@ def test_export_json():
     assert type(json_obj) == dict
     assert json_obj['num_feature'] == 4
     assert json_obj['task_type'] == 'kBinaryClfRegr'
-    assert json_obj['average_tree_output'] == True
+    assert json_obj['average_tree_output']
     assert json_obj['task_param'] == {'grove_per_class': False, 'leaf_vector_size': 1, 'num_class': 1,
                                       'output_type': 'float'}
     assert json_obj['model_param'] == {'global_bias': 0.0, 'pred_transform': 'identity'}