Muyi 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

muyi/__init__.py

@@ -0,0 +1,35 @@
+# utils
+from .utils import (
+    bcolors,
+    color_print,
+    save_pic_iterly,
+    read_csv_tqdm,
+    save_result_csv,
+    get_unique_save_path,
+)
+
+# gpu
+from .gpu import (
+    get_gpu_memory_usage,
+    display_gpu_memory_usage,
+)
+
+# graph
+from .graph import (
+    pyg_data_to_dgl_graph,
+)
+
+__all__ = [
+    # utils
+    'bcolors',
+    'color_print',
+    'save_pic_iterly',
+    'read_csv_tqdm',
+    'save_result_csv',
+    'get_unique_save_path',
+    # gpu
+    'get_gpu_memory_usage',
+    'display_gpu_memory_usage',
+    # graph
+    'pyg_data_to_dgl_graph',
+]

muyi/gpu.py

@@ -1,6 +1,5 @@
-import GPUtil
-
 def get_gpu_memory_usage():
+    import GPUtil
     # 获取GPU使用情况
     gpus = GPUtil.getGPUs()
     gpu_memory_info = []

muyi/graph.py

@@ -1,7 +1,6 @@
-import torch
-import dgl
-
 def pyg_data_to_dgl_graph(pyg_data_obj):
+    import torch
+    import dgl
     print(pyg_data_obj)
 
     # 获取边索引

muyi/utils.py

@@ -1,13 +1,69 @@
 import os
-import matplotlib.pyplot as plt
-import pandas as pd
-from tqdm import tqdm
+import csv
 import sys
 
-def color_print(content):
-    print(f'\033[1;46m{content}\033[0m\n')
+class bcolors:
+    HEADER = '\033[95m'    # 紫色，用于标题
+    OKBLUE = '\033[94m'    # 蓝色，用于正常或信息性消息
+    OKCYAN = '\033[96m'    # 青色（浅蓝色），用于信息性消息
+    OKGREEN = '\033[92m'   # 绿色，用于成功或确认的消息
+    WARNING = '\033[93m'   # 黄色，用于警告或重要提示
+    FAIL = '\033[91m'      # 红色，用于错误或失败的消息
+    ENDC = '\033[0m'       # 重置所有样式，回到默认颜色
+    BOLD = '\033[1m'       # 粗体文本
+    UNDERLINE = '\033[4m'  # 下划线文本
+    BLACK = '\033[30m'     # 黑色
+    RED = '\033[31m'       # 红色
+    GREEN = '\033[32m'     # 绿色
+    YELLOW = '\033[33m'    # 黄色
+    BLUE = '\033[34m'      # 蓝色
+    MAGENTA = '\033[35m'   # 品红色（紫红色）
+    CYAN = '\033[36m'      # 青色
+    WHITE = '\033[37m'     # 白色
+    # 添加背景颜色
+    BG_BLACK = '\033[40m'  # 黑色背景
+    BG_RED = '\033[41m'    # 红色背景
+    BG_GREEN = '\033[42m'  # 绿色背景
+    BG_YELLOW = '\033[43m' # 黄色背景
+    BG_BLUE = '\033[44m'   # 蓝色背景
+    BG_MAGENTA = '\033[45m'# 品红色（紫红色）背景
+    BG_CYAN = '\033[46m'   # 青色背景
+    BG_WHITE = '\033[47m'  # 白色背景
+
+
+def color_print(content, font_color='white', bg_color='bg_blue'):
+    colors = {
+        'header': bcolors.HEADER,
+        'okblue': bcolors.OKBLUE,
+        'okcyan': bcolors.OKCYAN,
+        'okgreen': bcolors.OKGREEN,
+        'warning': bcolors.WARNING,
+        'fail': bcolors.FAIL,
+        'black': bcolors.BLACK,
+        'red': bcolors.RED,
+        'green': bcolors.GREEN,
+        'yellow': bcolors.YELLOW,
+        'blue': bcolors.BLUE,
+        'magenta': bcolors.MAGENTA,
+        'cyan': bcolors.CYAN,
+        'white': bcolors.WHITE,
+        'bg_black': bcolors.BG_BLACK,
+        'bg_red': bcolors.BG_RED,
+        'bg_green': bcolors.BG_GREEN,
+        'bg_yellow': bcolors.BG_YELLOW,
+        'bg_blue': bcolors.BG_BLUE,
+        'bg_magenta': bcolors.BG_MAGENTA,
+        'bg_cyan': bcolors.BG_CYAN,
+        'bg_white': bcolors.BG_WHITE,
+        'end': bcolors.ENDC,
+    }
+
+    # Apply the background color first, then the font color
+    print(f'{colors[bg_color]}{colors[font_color]}{content}{colors["end"]}\n')
 
 def save_pic_iterly(pic_name, postfix, info):
+    import matplotlib.pyplot as plt
+
     pic_idx=1
     pic_name_full=f'{pic_name}_{pic_idx}.{postfix}'
 
@@ -21,6 +77,9 @@ def save_pic_iterly(pic_name, postfix, info):
     color_print(f'!!!!! {info} is saved in file {pic_name_full}')
 
 def read_csv_tqdm(path, **kwargs):
+    import pandas as pd
+    from tqdm import tqdm
+
     INPUT_FILENAME = path
     LINES_TO_READ_FOR_ESTIMATION = 20
     CHUNK_SIZE_PER_ITERATION = 10**5
@@ -44,3 +103,68 @@ def read_csv_tqdm(path, **kwargs):
     
     del df            
     return data
+
+def save_result_csv(csv_path, data):
+    """保存结果到CSV文件（通用型）
+
+    Args:
+        csv_path: CSV文件路径
+        data: dict, 包含要保存的数据，key为列名，value为对应的值
+    """
+    file_exists = os.path.exists(csv_path)
+    headers = list(data.keys())
+    values = list(data.values())
+
+    with open(csv_path, 'a', newline='') as f:
+        writer = csv.writer(f)
+        if not file_exists:
+            writer.writerow(headers)
+        writer.writerow(values)
+
+    print(f"Results saved to {csv_path}")
+
+def get_unique_save_path(folder_path, base_name_pattern, start_no=1):
+    """生成不重复的保存路径，自动递增版本号。
+
+    通过检查文件是否存在，自动递增编号直到找到一个不存在的文件路径。
+    如果目标文件夹不存在，会自动创建。
+
+    Args:
+        folder_path: 目标文件夹路径
+        base_name_pattern: 文件名模板，必须包含 {no} 占位符用于插入编号
+            例如: "result_{no}.csv", "model_v{no}.pt"
+        start_no: 起始编号，默认为 1
+
+    Returns:
+        str: 不重复的完整文件路径
+
+    Examples:
+        >>> get_unique_save_path("./output", "result_{no}.csv")
+        './output/result_1.csv'  # 如果不存在
+
+        >>> get_unique_save_path("./output", "result_{no}.csv")
+        './output/result_2.csv'  # 如果 result_1.csv 已存在
+
+        >>> get_unique_save_path("./models", "checkpoint_v{no}.pt", start_no=10)
+        './models/checkpoint_v10.pt'  # 从编号 10 开始
+
+        # 使用 f-string 动态构建模板（注意 {no} 需要用双花括号转义）
+        >>> model_name = "transformer"
+        >>> dataset = "ETTh1"
+        >>> get_unique_save_path("./results", f"{model_name}_{dataset}_{{no}}.csv")
+        './results/transformer_ETTh1_1.csv'
+
+        >>> pred_len = 96
+        >>> get_unique_save_path("./output", f"pred{pred_len}_exp{{no}}.npy")
+        './output/pred96_exp1.npy'
+    """
+    if not os.path.exists(folder_path):
+        os.makedirs(folder_path)
+
+    no = start_no
+    while True:
+        filename = base_name_pattern.format(no=no)
+        full_path = os.path.join(folder_path, filename)
+        if not os.path.exists(full_path):
+            return full_path
+        no += 1

muyi-0.0.9.dist-info/METADATA

@@ -0,0 +1,101 @@
+Metadata-Version: 2.4
+Name: Muyi
+Version: 0.0.9
+Summary: Some useful utils.
+Home-page: https://github.com/Muyiiiii/muyi
+Author: muyiiiii
+Author-email: 
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Provides-Extra: plot
+Requires-Dist: matplotlib; extra == "plot"
+Provides-Extra: csv
+Requires-Dist: pandas; extra == "csv"
+Requires-Dist: tqdm; extra == "csv"
+Provides-Extra: gpu
+Requires-Dist: GPUtil; extra == "gpu"
+Provides-Extra: graph
+Requires-Dist: torch; extra == "graph"
+Requires-Dist: dgl; extra == "graph"
+Provides-Extra: all
+Requires-Dist: matplotlib; extra == "all"
+Requires-Dist: pandas; extra == "all"
+Requires-Dist: tqdm; extra == "all"
+Requires-Dist: GPUtil; extra == "all"
+Requires-Dist: torch; extra == "all"
+Requires-Dist: dgl; extra == "all"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-python
+Dynamic: summary
+
+# muyi
+
+Some useful utils for GNNs and Deep Learning.
+
+## Installation
+
+```bash
+pip install muyi              # basic installation
+pip install muyi[plot]        # + matplotlib
+pip install muyi[csv]         # + pandas, tqdm
+pip install muyi[gpu]         # + GPUtil
+pip install muyi[graph]       # + torch, dgl
+pip install muyi[all]         # all dependencies
+```
+
+## utils
+
+1. `color_print(content, font_color, bg_color)`
+2. `save_pic_iterly(pic_name, postfix, info)`
+3. `read_csv_tqdm(path, **kwargs)`
+4. `save_result_csv(csv_path, data)`
+5. `get_unique_save_path(folder_path, base_name_pattern, start_no)`
+
+## graph
+
+1. `pyg_data_to_dgl_graph(pyg_data_obj)`
+
+## gpu
+
+1. `get_gpu_memory_usage()`
+2. `display_gpu_memory_usage()`
+
+## Upload to PyPI
+
+```bash
+# 1. Install build tools
+pip install build twine
+
+# 2. Build package
+python -m build
+
+# 3. Upload to PyPI
+twine upload dist/*
+# Username: __token__
+# Password: your PyPI API Token (starts with pypi-)
+```
+
+Get API Token: https://pypi.org/manage/account/token/
+
+### Save Token Locally (Optional)
+
+Create `~/.pypirc` file to avoid entering credentials each time:
+
+```ini
+[pypi]
+username = __token__
+password = pypi-your-token-here
+```
+
+- **Linux/macOS**: `~/.pypirc`
+- **Windows**: `C:\Users\<username>\.pypirc`

muyi-0.0.9.dist-info/RECORD

@@ -0,0 +1,9 @@
+muyi/__init__.py,sha256=tY99oEmStngLIJCZN1QrKR6oOce-8zt_dzW3hE4SbAU,558
+muyi/gpu.py,sha256=wQozizgnFOCtKThimSQZnArJLeQD485H6mEuAJT84yE,678
+muyi/graph.py,sha256=tOkXMjtOzxxIB326FOshhcZYf4d_6BwPjrxURqywz90,1151
+muyi/utils.py,sha256=Ik9rf6EXgluLnr__SMiHERyWxV3J3UZmVvUdc36XF8A,6066
+muyi-0.0.9.dist-info/licenses/LICENSE,sha256=GzHS9pi6vw_oGwKIGxAPch8iuVrHazOilEsYL7AlPK0,1085
+muyi-0.0.9.dist-info/METADATA,sha256=nIDB8VkyRXX4vokOSkSmlc4vrAXzAEhy0K7mPkhz6bo,2506
+muyi-0.0.9.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+muyi-0.0.9.dist-info/top_level.txt,sha256=3X0xmx0SvmCNEj1ISqmD2evKh_ejCJQaLA4GEbGfKc4,5
+muyi-0.0.9.dist-info/RECORD,,

{Muyi-0.0.7.dist-info → muyi-0.0.9.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (70.2.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

Muyi-0.0.7.dist-info/METADATA

@@ -1,36 +0,0 @@
-Metadata-Version: 2.1
-Name: Muyi
-Version: 0.0.7
-Summary: Some useful utils.
-Home-page: https://github.com/Muyiiiii/muyi
-Author: muyiiiii
-Author-email: 
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3
-Description-Content-Type: text/markdown
-License-File: LICENSE
-
-# muyi
-
-Some useful utils for GNNs and Deep Learning.
-
-```bash
-pip install muyi
-```
-
-## utils
-
-1. `color_print(content)`
-2. `save_pic_iterly(pic_name, postfix, info)`
-3. `read_csv_tqdm(path, **kwargs)`
-
-## graph
-
-1. `pyg_data_to_dgl_graph(pyg_data_obj)`
-
-## gpu
-
-1. `get_gpu_memory_usage()`
-2. `display_gpu_memory_usage()`

Muyi-0.0.7.dist-info/RECORD

@@ -1,9 +0,0 @@
-muyi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-muyi/gpu.py,sha256=JLr-hQ84gv7afgFX6XHaquvN-HB19NqvwHxoq-rugdM,676
-muyi/graph.py,sha256=4okLvgp0TboSpuKtlzXTTbLin2K-RMVxhqJ7CZt6Tps,1145
-muyi/utils.py,sha256=XJaNLLNvSbaCcFm_dVnHS8RG4AGnQoJj0h5nt5tSYJk,1429
-Muyi-0.0.7.dist-info/LICENSE,sha256=GzHS9pi6vw_oGwKIGxAPch8iuVrHazOilEsYL7AlPK0,1085
-Muyi-0.0.7.dist-info/METADATA,sha256=LeDChXIF81CatUgGGynldsi5F8PD8IXfiUMvEwZ70_Q,741
-Muyi-0.0.7.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-Muyi-0.0.7.dist-info/top_level.txt,sha256=3X0xmx0SvmCNEj1ISqmD2evKh_ejCJQaLA4GEbGfKc4,5
-Muyi-0.0.7.dist-info/RECORD,,