File size: 1,706 Bytes
b8c8435 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 | """
UIR Package Setup
"""
from setuptools import setup, find_packages
with open("README.md", "r", encoding="utf-8") as f:
long_description = f.read()
setup(
name="uir",
version="0.1.0",
author="ViGiL Project",
description="Unified Instruction Representation for Heterogeneous Malware Analysis",
long_description=long_description,
long_description_content_type="text/markdown",
packages=find_packages(),
python_requires=">=3.8",
install_requires=[
"pefile>=2023.2.7",
"python-magic-bin>=0.4.14",
"py7zr>=0.20.8",
"rarfile>=4.1",
"pycdlib>=1.14.0",
"oletools>=0.60.1",
"torch>=2.0.0",
"networkx>=3.0",
"tqdm>=4.65.0",
"pydantic>=2.0.0",
"numpy>=1.24.0",
"scikit-learn>=1.3.0",
"pylnk3>=0.4.2",
],
extras_require={
"gpu": ["torch-geometric>=2.4.0", "orjson>=3.9.0"],
"m4": ["msgpack>=1.0.5", "orjson>=3.9.0"],
"fast": ["orjson>=3.9.0", "msgpack>=1.0.5"],
"dev": ["pytest", "black", "flake8"],
},
entry_points={
"console_scripts": [
"uir=uir.pipeline.cli:main",
],
},
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Topic :: Security",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
],
)
|