Fix graphql

ukaea · Jun 19, 2024 · 19ef1db · 19ef1db
1 parent a1178ab
commit 19ef1db
Showing 1 changed file with 8 additions and 10 deletions.
diff --git a/src/api/graphql.py b/src/api/graphql.py
@@ -260,21 +260,19 @@ def on_request_end(self):
 )
 class Shot:
     @strawberry.field
-    def signal_datasets(
-        self,
-        limit: Optional[int] = None,
-        where: Optional[ShotWhereFilter] = None,
-    ) -> List[strawberry.LazyType["Shot", __module__]]: # noqa: F821
-        results = do_where_child_member(self.signal_datasets, where)
+    def signals(
+        self, limit: Optional[int] = None, where: Optional[SignalWhereFilter] = None
+    ) -> List[strawberry.LazyType["Signal", __module__]]:  # noqa: F821
+        results = do_where_child_member(self.signals, where)
         if limit is not None:
             results = results[:limit]
         return results
 
     @strawberry.field
-    def signals(
-        self, limit: Optional[int] = None, where: Optional[SignalWhereFilter] = None
-    ) -> List[strawberry.LazyType["Signal", __module__]]: # noqa: F821
-        results = do_where_child_member(self.signals, where)
+    def sources(
+        self, limit: Optional[int] = None, where: Optional[SourceWhereFilter] = None
+    ) -> List[strawberry.LazyType["Source", __module__]]:  # noqa: F821
+        results = do_where_child_member(self.sources, where)
         if limit is not None:
             results = results[:limit]
         return results