[BUG] swanlab.config can't fit hydra - is not JSON serializable

[Zeyi-Lin]

🤩 Features description [Please make everyone to understand it]

test code:

import hydra
from omegaconf import DictConfig, OmegaConf
import json
import swanlab


@hydra.main(config_path=".", config_name="configs", version_base="1.3.2")
def startup(cfg: DictConfig):
    config = dict(cfg)
    print(config)
    print("#" * 30)
    swanlab.init(experiment_name="abc", config=config)
   
startup()

configs.yaml, locate in the same dir:

data:
  dir_image: "./data/samimage256/256/"
  dir_mask: "./data/samimage256/mymask/"
  image_size: 512

hardware:
  seed: 2021
  num_workers: 4

Error:

Traceback (most recent call last):
  File "/Users/zeyilin/miniforge3/envs/test_py310/lib/python3.10/site-packages/swanlab/data/run/main.py", line 310, in __check_config
    ujson.dumps(config)
TypeError: {'dir_image': './data/samimage256/256/', 'dir_mask': './data/samimage256/mymask/', 'image_size': 512} is not JSON serializable

[Zeyi-Lin]

If use following code, run will ok:

import hydra
from omegaconf import DictConfig, OmegaConf
import swanlab


@hydra.main(config_path=".", config_name="configs", version_base="1.3.2")
def startup(cfg: DictConfig):
    config = OmegaConf.to_container(cfg, resolve=True)
    config = dict(config)
    print(config)
    print("#" * 30)
    swanlab.init(experiment_name="abc", config=config)

startup()

ujson.dumps can't process omegaconf.DictConfig directly, so swanlab.config need fit some mainstream tools's config methods with preprocess

[Zeyi-Lin]

Fix: #238
Use a general json preprocess function to fix this problem.
Maybe in the future, we should test more framework'config method to make the preprocess method better 😄

def __check_config(self, config: dict) -> dict:
    """检查实验配置是否合法"""
    if config is None:
        return {}
    # config必须可以被json序列化
    try:
        check_config = ujson.dumps(json_serializable(dict(config)))
    except:
        raise TypeError(f"config: {config} is not a valid dict, which can be json serialized")
    return check_config

function json_serializable:

def json_serializable(obj: dict):
    import datetime

    """
    Convert an object into a JSON-serializable format.
    """
    if isinstance(obj, (int, float, str, bool, type(None))):
        return obj

    if isinstance(obj, (datetime.date, datetime.datetime)):
        return obj.isoformat()

    if isinstance(obj, (list, tuple)):
        return [json_serializable(item) for item in obj]

    if isinstance(obj, dict):
        return {key: json_serializable(value) for key, value in obj.items()}

    return str(obj)