prepare_custom_config_dict = {
          # user will manually define the corresponding observed
          # module class which has a from_float class method that converts
          # float custom module to observed custom module
          "float_to_observed_custom_module_class": {
              CustomModule: ObservedCustomModule
          }
       }