Making sure input_dims set converted to list.

mlp/data_providers.py

@@ -284,6 +284,8 @@ class CCPPDataProvider(DataProvider):
             assert input_dims.issubset({0, 1, 2, 3}), (
                 'input_dims should be a subset of {0, 1, 2, 3}'
             )
+            # convert to list as cannot index ndarray by set
+            input_dims = list(input_dims)
         loaded = np.load(data_path)
         inputs = loaded[which_set + '_inputs']
         if input_dims is not None: