Address comments.

Signed-off-by: Qing Wang <kingchin1218@gmail.com>
ray-project · Jul 11, 2023 · 5c2fba0 · 5c2fba0
1 parent 992aac6
commit 5c2fba0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/fed/_private/fed_call_holder.py b/fed/_private/fed_call_holder.py
@@ -56,7 +56,7 @@ def options(self, **options):
  return self
 
  def internal_remote(self, *args, **kwargs):
- if self._node_party is None or len(self._node_party) == 0:
+ if not self._node_party:
  raise ValueError("You should specify a party name on the fed actor.")
 
  # Generate a new fed task id for this call.

diff --git a/fed/api.py b/fed/api.py
@@ -300,9 +300,9 @@ def options(self, **options):
  return self
 
  def remote(self, *args, **kwargs):
- assert (
- self._node_party is not None
- ), "A fed function should be specified within a party to execute."
+ if not self._node_party:
+ raise ValueError("You should specify a party name on the fed function.")
+
  return self._fed_call_holder.internal_remote(*args, **kwargs)
 
  def _execute_impl(self, args, kwargs):