From b39b86309cdaf8a6d8b207b3278781fd80a47140 Mon Sep 17 00:00:00 2001 From: Nil-Andreu Date: Mon, 9 Jan 2023 09:03:02 +0100 Subject: [PATCH] [FIX] Import on top --- backend/oasst_backend/utils/hugging_face.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/oasst_backend/utils/hugging_face.py b/backend/oasst_backend/utils/hugging_face.py index 80be074d..ef440055 100644 --- a/backend/oasst_backend/utils/hugging_face.py +++ b/backend/oasst_backend/utils/hugging_face.py @@ -2,6 +2,7 @@ from enum import Enum from typing import Any, Dict import aiohttp +from loguru import logger from oasst_backend.config import settings from oasst_shared.exceptions import OasstError, OasstErrorCode @@ -51,7 +52,6 @@ class HuggingFaceAPI: async with session.post(self.api_url, headers=self.headers, json=payload) as response: # If we get a bad response if response.status != 200: - from loguru import logger logger.error(response) logger.info(self.headers)