Fixed a doc error

NVIDIA · Jan 23, 2025 · 260104b · 260104b
1 parent a3cc343
commit 260104b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/examples/advanced/streaming/README.md b/examples/advanced/streaming/README.md
@@ -37,7 +37,7 @@ Currently, following implementations are available,
 To use ContainerRetriever, the container must be given a name and added on the sending site,
 
 ```
-ContainerRetriever.add("model", model_dict)
+ContainerRetriever.add_container("model", model_dict)
 ```
 
 ## Example Jobs

diff --git a/nvflare/app_common/streamers/container_streamer.py b/nvflare/app_common/streamers/container_streamer.py
@@ -54,6 +54,8 @@ def consume(
             if isinstance(self.container, dict):
                 key, value = entry
                 self.container[key] = value
+            elif isinstance(self.container, set):
+                self.container.add(entry)
             else:
                 self.container.append(entry)
         except Exception as ex: