Minor corrections and reverts

Signed-off-by: noopur <[email protected]>

tests/end_to_end/utils/federation_helper.py

@@ -136,7 +136,7 @@ def _create_tarball(collaborator_name, index, local_bind_path, add_data):
                 ]
                 client_certs = " ".join(client_cert_entries) if client_cert_entries else ""
                 tarfiles += f" agg_to_col_{collaborator_name}_signed_cert.zip {client_certs}"
-                # IMPORTANT: Model xgb_higgs uses format like data/1 and data/2, thus adding data to tarball in the same format.
+                # IMPORTANT: Model XGBoost(xgb_higgs) uses format like data/1 and data/2, thus adding data to tarball in the same format.
                 if add_data:
                     tarfiles += f" data/{index}"