avoid pydantic warning for gradient_fn field

AgnostiqHQ · Sep 12, 2023 · 7e4906e · 7e4906e
1 parent 8d5d509
commit 7e4906e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/covalent/_shared_files/qinfo.py b/covalent/_shared_files/qinfo.py
@@ -44,7 +44,7 @@ class QNodeSpecs(BaseModel):
     expansion_strategy: str
     gradient_options: Dict[str, int]
     interface: Optional[str]
-    gradient_fn: Optional[str]
+    gradient_fn: Any  # can be string or `qml.gradients.gradient_transform`
     num_gradient_executions: Any = 0
     num_parameter_shift_executions: int = None