minor changes

Author: najiyacl

ads/feature_store/dataset.py

@@ -529,6 +529,8 @@ def add_models(self, model_details: ModelDetails) -> "Dataset":
                 if model_id not in updated_model_details["items"]:
                     logger.warning(f"Either model with Id '{model_id}' doesnt exist or unable to validate")
 
+        return self
+
     def remove_models(self, model_details: ModelDetails) -> "Dataset":
         """remove model details from the dataset, remove from the existing dataset model id list