Merge pull request #275 from loayei/fix-logging

fix-logger

Author: mccrindlebrian

folding/base/miner.py

@@ -26,7 +26,7 @@
 from folding.base.neuron import BaseNeuron
 from folding.protocol import PingSynapse
 from folding.utils.config import add_miner_args
-from loguru import logger
+from folding.utils.logger import logger
 
 
 class BaseMinerNeuron(BaseNeuron):

folding/base/neuron.py

@@ -30,7 +30,7 @@
 from folding import __OPENMM_VERSION_TAG__
 from folding.utils.ops import OpenMMException, load_pkl, write_pkl
 from folding.mock import MockSubtensor, MockMetagraph
-from loguru import logger
+from folding.utils.logger import logger
 
 
 class BaseNeuron(ABC):

folding/base/simulation.py

@@ -8,7 +8,7 @@
 
 import bittensor as bt
 from folding.utils.opemm_simulation_config import SimulationConfig
-from loguru import logger
+from folding.utils.logger import logger
 
 
 class GenericSimulation(ABC):

folding/base/validator.py

@@ -25,7 +25,7 @@
 import threading
 import bittensor as bt
 from pathlib import Path
-from loguru import logger
+from folding.utils.logger import logger
 
 from typing import List, Optional
 

folding/experiments/sensitivity_analysis.py

@@ -5,7 +5,7 @@
 import time
 import wandb
 from tqdm import tqdm
-from loguru import logger
+from folding.utils.logger import logger
 from pathlib import Path
 from typing import Dict, List
 

folding/miners/folding_miner.py

@@ -26,7 +26,7 @@
     write_pkl,
 )
 from folding.utils.opemm_simulation_config import SimulationConfig
-from loguru import logger
+from folding.utils.logger import logger
 
 # root level directory for the project (I HATE THIS)
 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

folding/mock.py

@@ -5,7 +5,7 @@
 import bittensor as bt
 
 from typing import List
-from loguru import logger
+from folding.utils.logger import logger
 
 
 class MockSubtensor(bt.MockSubtensor):

folding/organic/validator.py

@@ -3,7 +3,7 @@
 import random
 import asyncio
 import bittensor as bt
-from loguru import logger
+from folding.utils.logger import logger
 
 from typing import Any, Literal, Union, Tuple
 

folding/protocol.py

@@ -19,7 +19,7 @@
 import typing
 import base64
 import bittensor as bt
-from loguru import logger
+from folding.utils.logger import logger
 
 from folding.utils.opemm_simulation_config import SimulationConfig
 

folding/rewards/reward_pipeline.py

@@ -2,7 +2,7 @@
 import bittensor as bt
 from folding.store import Job
 from folding.rewards.linear_reward import divide_decreasing
-from loguru import logger
+from folding.utils.logger import logger
 
 
 async def reward_pipeline(