Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion machine_learning_hep/analysis/analyzer.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
from os import makedirs
from os.path import exists, join

from machine_learning_hep.io import dump_yaml_from_dict
from machine_learning_hep.io_ml_utils import dump_yaml_from_dict

# HF specific imports
from machine_learning_hep.workflow.workflow_base import WorkflowBase
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/analysis/systematics.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
from ROOT import TCanvas, TFile, TGraphErrors, TLegend, kAzure, kBlack, kBlue, kGreen, kOrange, kRed, kViolet, kYellow

from machine_learning_hep.fitting.helpers import MLFitter
from machine_learning_hep.io import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.io_ml_utils import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.logger import get_logger
from machine_learning_hep.multiprocesser import MultiProcesser
from machine_learning_hep.utilities_plot import load_root_style
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/fitting/simple_fit.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@

from machine_learning_hep.fitting.fitters import FitAliHF, FitROOTGauss
from machine_learning_hep.fitting.utils import save_fit
from machine_learning_hep.io import parse_yaml
from machine_learning_hep.io_ml_utils import parse_yaml
from machine_learning_hep.logger import configure_logger # , get_logger

#############################################################################
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/fitting/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@
# pylint: disable=import-error, no-name-in-module, unused-import
from ROOT import TFile

from machine_learning_hep.io import checkdir, dump_yaml_from_dict, parse_yaml
from machine_learning_hep.io_ml_utils import checkdir, dump_yaml_from_dict, parse_yaml
from machine_learning_hep.logger import get_logger


Expand Down
24 changes: 13 additions & 11 deletions machine_learning_hep/io.py → machine_learning_hep/io_ml_utils.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,16 @@
# © Copyright CERN 2018. All rights not expressly granted are reserved. #
# Author: Gian.Michele.Innocenti@cern.ch #
# This program is free software: you can redistribute it and/or modify it #
# under the terms of the GNU General Public License as published by the #
# Free Software Foundation, either version 3 of the License, or (at your #
# option) any later version. This program is distributed in the hope that #
# it will be useful, but WITHOUT ANY WARRANTY; without even the implied #
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. #
# See the GNU General Public License for more details. #
# You should have received a copy of the GNU General Public License #
# along with this program. if not, see <https://www.gnu.org/licenses/>. #
#############################################################################
## © Copyright CERN 2024. All rights not expressly granted are reserved. ##
## Author: Gian.Michele.Innocenti@cern.ch ##
## This program is free software: you can redistribute it and/or modify it ##
## under the terms of the GNU General Public License as published by the ##
## Free Software Foundation, either version 3 of the License, or (at your ##
## option) any later version. This program is distributed in the hope that ##
## it will be useful, but WITHOUT ANY WARRANTY; without even the implied ##
## warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. ##
## See the GNU General Public License for more details. ##
## You should have received a copy of the GNU General Public License ##
## along with this program. if not, see <https://www.gnu.org/licenses/>. ##
#############################################################################

"""
Methods to: manage input/output
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/multiprocesser.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
import os
import tempfile

from machine_learning_hep.io import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.io_ml_utils import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.logger import get_logger
from machine_learning_hep.utilities import merge_method, mergerootfiles

Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/optimisation/bayesian_opt.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@
from yaml.representer import RepresenterError

# from shap.plots.colors import red_blue as shap_cmap_red_blue
from machine_learning_hep.io import dict_yamlable, dump_yaml_from_dict, parse_yaml
from machine_learning_hep.io_ml_utils import dict_yamlable, dump_yaml_from_dict, parse_yaml

# Change to that backend to not have problems with saving fgures
# when X11 connection got lost
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/optimisation/grid_search.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
import pandas as pd
from sklearn.model_selection import GridSearchCV

from machine_learning_hep.io import dump_yaml_from_dict, parse_yaml, print_dict
from machine_learning_hep.io_ml_utils import dump_yaml_from_dict, parse_yaml, print_dict
from machine_learning_hep.logger import get_logger
from machine_learning_hep.models import savemodels
from machine_learning_hep.optimisation.metrics import get_scorers
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/optimiser.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@
vardistplot,
vardistplot_probscan,
)
from machine_learning_hep.io import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.io_ml_utils import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.logger import get_logger
from machine_learning_hep.models import (
apply,
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/processer.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@
import uproot
from pandas.api.types import is_numeric_dtype

from .io import dump_yaml_from_dict
from .io_ml_utils import dump_yaml_from_dict
from .logger import get_logger
from .utilities import (
count_df_length_pkl,
Expand Down
2 changes: 1 addition & 1 deletion machine_learning_hep/utilities_plot.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@
kWhite,
)

from machine_learning_hep.io import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.io_ml_utils import dump_yaml_from_dict, parse_yaml
from machine_learning_hep.logger import get_logger


Expand Down
Loading