def METHOD_NAME(self):
    self.num_dropped_rows = None
    self.model_uri = None
    self.model_details = None
    self.version = None
    self.register_model_name = self.step_config.get("model_name")
    if self.register_model_name is None:
        raise MlflowException(
            "Missing 'model_name' config in register step config.",
            error_code=INVALID_PARAMETER_VALUE,
        )
    self.allow_non_validated_model = self.step_config.get("allow_non_validated_model", False)
    self.registry_uri = self.step_config.get("registry_uri", None)