diff --git a/README.md b/README.md index b783041..d922e3c 100644 --- a/README.md +++ b/README.md @@ -92,8 +92,8 @@ point. The necessary data will be: - Costs: The total cost per media unit per channel. ```python -# Let's assume we have the following datasets with the following shapes (we use -the `simulate_dummy_data` function in utils for this example): +# Let's assume we have the following datasets with the following shapes (we use the +# `simulate_dummy_data` function in utils for this example): media_data, extra_features, target, costs = utils.simulate_dummy_data( data_size=160, n_media_channels=3,