From 8c196ad8f31d8cbb83b5f2346f02eea4b3aa3808 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 12 Jan 2023 23:51:33 -0600 Subject: [PATCH] fix(gui): invalidate loading query after mutations --- gui/src/components/Img2Img.tsx | 2 +- gui/src/components/Inpaint.tsx | 4 +--- gui/src/components/Txt2Img.tsx | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/gui/src/components/Img2Img.tsx b/gui/src/components/Img2Img.tsx index 123d6401..62700503 100644 --- a/gui/src/components/Img2Img.tsx +++ b/gui/src/components/Img2Img.tsx @@ -36,7 +36,7 @@ export function Img2Img(props: Img2ImgProps) { const client = mustExist(useContext(ClientContext)); const query = useQueryClient(); const upload = useMutation(uploadSource, { - onSuccess: () => query.invalidateQueries({ queryKey: 'ready '}), + onSuccess: () => query.invalidateQueries({ queryKey: 'ready' }), }); const state = mustExist(useContext(StateContext)); diff --git a/gui/src/components/Inpaint.tsx b/gui/src/components/Inpaint.tsx index 60580470..8685dffa 100644 --- a/gui/src/components/Inpaint.tsx +++ b/gui/src/components/Inpaint.tsx @@ -141,12 +141,10 @@ export function Inpaint(props: InpaintProps) { const setInpaint = useStore(state, (s) => s.setInpaint); // eslint-disable-next-line @typescript-eslint/unbound-method const setLoading = useStore(state, (s) => s.setLoading); - // eslint-disable-next-line @typescript-eslint/unbound-method - const pushHistory = useStore(state, (s) => s.pushHistory); const query = useQueryClient(); const upload = useMutation(uploadSource, { - onSuccess: () => query.invalidateQueries({ queryKey: 'ready '}), + onSuccess: () => query.invalidateQueries({ queryKey: 'ready' }), }); // eslint-disable-next-line no-null/no-null const canvasRef = useRef(null); diff --git a/gui/src/components/Txt2Img.tsx b/gui/src/components/Txt2Img.tsx index e293fa65..75fd71fb 100644 --- a/gui/src/components/Txt2Img.tsx +++ b/gui/src/components/Txt2Img.tsx @@ -34,7 +34,7 @@ export function Txt2Img(props: Txt2ImgProps) { const client = mustExist(useContext(ClientContext)); const query = useQueryClient(); const generate = useMutation(generateImage, { - onSuccess: () => query.invalidateQueries({ queryKey: 'ready '}), + onSuccess: () => query.invalidateQueries({ queryKey: 'ready' }), }); const state = mustExist(useContext(StateContext));