yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-10-31 09:50:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove postponed eval of annotations from automl.py

Browse Source
This commit is contained in:
Andrea W 2023-01-27 10:00:16 +00:00
parent bdbba6e3bc
commit 2f93736740
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
flaml/automl/automl.py
View File

@ -2,7 +2,6 @@
# * Copyright (c) FLAML authors. All rights reserved. # * Copyright (c) FLAML authors. All rights reserved.
# * Licensed under the MIT License. See LICENSE file in the # * Licensed under the MIT License. See LICENSE file in the
# * project root for license information. # * project root for license information.
from __future__ import annotations
import time import time
import os import os
import sys import sys
@ -344,10 +343,7 @@ class AutoMLState:
return sampled_X_train, sampled_y_train, sampled_weight, groups return sampled_X_train, sampled_y_train, sampled_weight, groups
@staticmethod @staticmethod
def _compute_with_config_base(config_w_resource, def _compute_with_config_base(config_w_resource, state, estimator, is_report=True):
state: AutoMLState,
estimator,
is_report=True):
if "FLAML_sample_size" in config_w_resource: if "FLAML_sample_size" in config_w_resource:
sample_size = int(config_w_resource["FLAML_sample_size"]) sample_size = int(config_w_resource["FLAML_sample_size"])
else: else: