From 43cd970100434a0029536ee125ab8c3271626ed8 Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Tue, 1 Nov 2016 11:11:22 +0800
Subject: [PATCH] make set_dataset and mutate_dataset async RPCs

---
 artiq/language/environment.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/artiq/language/environment.py b/artiq/language/environment.py
index 822279e48..626af2550 100644
--- a/artiq/language/environment.py
+++ b/artiq/language/environment.py
@@ -3,6 +3,7 @@ from inspect import isclass
 
 from artiq.protocols import pyon
 from artiq.language import units
+from artiq.language.core import rpc
 
 
 __all__ = ["NoDefault",
@@ -274,6 +275,7 @@ class HasEnvironment:
         kernel_invariants = getattr(self, "kernel_invariants", set())
         self.kernel_invariants = kernel_invariants | {key}
 
+    @rpc(flags={"async"})
     def set_dataset(self, key, value,
                     broadcast=False, persist=False, save=True):
         """Sets the contents and handling modes of a dataset.
@@ -290,6 +292,7 @@ class HasEnvironment:
         """
         self.__dataset_mgr.set(key, value, broadcast, persist, save)
 
+    @rpc(flags={"async"})
     def mutate_dataset(self, key, index, value):
         """Mutate an existing dataset at the given index (e.g. set a value at
         a given position in a NumPy array)