[pycue] Add missing exception parsing to setAllocation. (#1172)

AcademySoftwareFoundation · Oct 25, 2022 · 234a9ce · 234a9ce
1 parent 962b227
commit 234a9ce
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pycue/opencue/wrappers/host.py b/pycue/opencue/wrappers/host.py
@@ -19,6 +19,7 @@
 import time
 
 from opencue import Cuebot
+from opencue import util
 from opencue.compiled_proto import comment_pb2
 from opencue.compiled_proto import host_pb2
 import opencue.wrappers.comment
@@ -149,6 +150,7 @@ def renameTag(self, oldTag, newTag):
  host_pb2.HostRenameTagRequest(host=self.data, old_tag=oldTag, new_tag=newTag),
  timeout=Cuebot.Timeout)
 
+ @util.grpcExceptionParser
  def setAllocation(self, allocation):
  """Sets the host to the given allocation.