diff --git a/bsuite/logging/logging_utils.py b/bsuite/logging/logging_utils.py index 9e4c0773..84c73866 100644 --- a/bsuite/logging/logging_utils.py +++ b/bsuite/logging/logging_utils.py @@ -16,7 +16,7 @@ # ============================================================================ """Read functionality for local csv-based experiments.""" -import collections +import collections.abc import copy from typing import Any, Callable, List, Mapping, Sequence, Tuple, Union @@ -73,7 +73,7 @@ def load_multiple_runs( # Convert any inputs to dictionary format. if isinstance(path_collection, six.string_types): path_collection = {path_collection: path_collection} - if not isinstance(path_collection, collections.Mapping): + if not isinstance(path_collection, collections.abc.Mapping): path_collection = {path: path for path in path_collection} # Loop through multiple bsuite runs, and apply single_load_fn to each.