Muyi 0.0.8__tar.gz → 0.0.9__tar.gz

muyi-0.0.9/Muyi.egg-info/PKG-INFO

@@ -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.8 → muyi-0.0.9}/Muyi.egg-info/SOURCES.txt

@@ -5,6 +5,7 @@ setup.py
 Muyi.egg-info/PKG-INFO
 Muyi.egg-info/SOURCES.txt
 Muyi.egg-info/dependency_links.txt
+Muyi.egg-info/requires.txt
 Muyi.egg-info/top_level.txt
 muyi/__init__.py
 muyi/gpu.py

muyi-0.0.9/Muyi.egg-info/requires.txt

@@ -0,0 +1,22 @@
+
+[all]
+matplotlib
+pandas
+tqdm
+GPUtil
+torch
+dgl
+
+[csv]
+pandas
+tqdm
+
+[gpu]
+GPUtil
+
+[graph]
+torch
+dgl
+
+[plot]
+matplotlib

muyi-0.0.9/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,61 @@
+# 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/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-0.0.8 → muyi-0.0.9}/muyi/gpu.py

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

{muyi-0.0.8 → muyi-0.0.9}/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-0.0.8 → muyi-0.0.9}/muyi/utils.py

@@ -1,7 +1,5 @@
 import os
-import matplotlib.pyplot as plt
-import pandas as pd
-from tqdm import tqdm
+import csv
 import sys
 
 class bcolors:
@@ -64,6 +62,8 @@ def color_print(content, font_color='white', bg_color='bg_blue'):
     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}'
 
@@ -77,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
@@ -100,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.8 → muyi-0.0.9}/setup.py

@@ -3,7 +3,7 @@ with open("README.md", "r") as fh:
     long_description = fh.read()
 setuptools.setup(
     name="Muyi",  # 模块名称
-    version="0.0.8",  # 当前版本
+    version="0.0.9",  # 当前版本
     author="muyiiiii",  # 作者
     author_email="",  # 作者邮箱
     description="Some useful utils.",  # 模块简介
@@ -19,6 +19,22 @@ setuptools.setup(
     ],
     # 依赖模块
     install_requires=[
+        # 无核心依赖，所有依赖都是可选的
     ],
+    # 可选依赖
+    extras_require={
+        'plot': ['matplotlib'],           # save_pic_iterly
+        'csv': ['pandas', 'tqdm'],        # read_csv_tqdm
+        'gpu': ['GPUtil'],                # get_gpu_memory_usage
+        'graph': ['torch', 'dgl'],        # pyg_data_to_dgl_graph
+        'all': [                          # 安装全部
+            'matplotlib',
+            'pandas',
+            'tqdm',
+            'GPUtil',
+            'torch',
+            'dgl',
+        ],
+    },
     python_requires='>=3',
 )

muyi-0.0.8/Muyi.egg-info/PKG-INFO

@@ -1,36 +0,0 @@
-Metadata-Version: 2.1
-Name: Muyi
-Version: 0.0.8
-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.8/PKG-INFO

@@ -1,36 +0,0 @@
-Metadata-Version: 2.1
-Name: Muyi
-Version: 0.0.8
-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.8/README.md

@@ -1,22 +0,0 @@
-# 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()`