Skip to content

Commit

Permalink
remove conversion checker to avoid circular import issue
Browse files Browse the repository at this point in the history
Signed-off-by: kta-intel <[email protected]>
  • Loading branch information
kta-intel committed Nov 18, 2024
1 parent d65def1 commit 63be874
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions openfl/interface/aggregation_functions/fed_bagging.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
import numpy as np

from openfl.interface.aggregation_functions.core import AggregationFunction
from openfl.federated.task.runner_xgb import check_precision_loss

def get_global_model(iterator, target_round):
"""
Expand Down Expand Up @@ -128,6 +127,5 @@ def call(self, local_tensors, db_iterator, tensor_name, fl_round, *_):
global_model_bytes = global_model_json.encode("utf-8")

global_model_float32_array = np.frombuffer(global_model_bytes, dtype=np.uint8).astype(np.float32)
check_precision_loss(logger, global_model_float32_array, global_model)

return global_model_float32_array

0 comments on commit 63be874

Please sign in to comment.