diff --git a/deepcomp/env/util/utility.py b/deepcomp/env/util/utility.py index 3931019..6f06dd9 100644 --- a/deepcomp/env/util/utility.py +++ b/deepcomp/env/util/utility.py @@ -16,7 +16,7 @@ def linear_clipped_utility(curr_dr, max_dr=MAX_UTILITY): """ assert curr_dr >= 0 and max_dr >= 0 assert MIN_UTILITY == 0 and MAX_UTILITY == max_dr, \ - f"The chosen linear utility requires MIN_UTILITY=0 and sensible MAX_UTILITY. Set sensible values manually!" + "The chosen linear utility requires MIN_UTILITY=0 and sensible MAX_UTILITY. Set sensible values manually!" return np.clip(curr_dr, MIN_UTILITY, MAX_UTILITY) diff --git a/deepcomp/util/env_setup.py b/deepcomp/util/env_setup.py index d261e61..4799264 100644 --- a/deepcomp/util/env_setup.py +++ b/deepcomp/util/env_setup.py @@ -180,7 +180,7 @@ def get_env(map_size, bs_dist, num_static_ues, num_slow_ues, num_fast_ues, shari """Create and return the environment corresponding to the given map_size""" assert map_size in SUPPORTED_ENVS, f"Environment {map_size} is not one of {SUPPORTED_ENVS}." assert util_func in SUPPORTED_UTILITIES, \ - f"Utility function {util_func} not supported. Supported: {SUPPORTED_UTILITIES}" + f"Utility function {util_func} not supported. Supported: {SUPPORTED_UTILITIES}" # create map and BS list map, bs_list = None, None