wisent 0.7.925__tar.gz

wisent-0.7.925/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023-2024 Wisent Team
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE. 

wisent-0.7.925/PKG-INFO

@@ -0,0 +1,68 @@
+Metadata-Version: 2.4
+Name: wisent
+Version: 0.7.925
+Summary: Monitor and influence AI Brains
+Home-page: https://github.com/wisent-ai/wisent
+Author: Lukasz Bartoszcze and the Wisent Team
+Author-email: lukasz.bartoszcze@wisent.ai
+Keywords: nlp,machine learning,language models,safety,guardrails,lm-evaluation-harness
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=1.9.0
+Requires-Dist: transformers>=4.20.0
+Requires-Dist: tqdm>=4.50.0
+Requires-Dist: scikit-learn>=0.24.0
+Requires-Dist: pandas>=1.2.0
+Requires-Dist: numpy>=1.21.0
+Requires-Dist: datasets>=2.0.0
+Requires-Dist: sentence-transformers>=2.0.0
+Requires-Dist: faiss-cpu>=1.7.0
+Requires-Dist: uncensorbench>=0.2.0
+Requires-Dist: pebble>=5.0.0
+Requires-Dist: latex2sympy2_extended>=1.0.0
+Requires-Dist: sae_lens>=0.1.0
+Requires-Dist: trl>=0.7.0
+Provides-Extra: harness
+Requires-Dist: lm-eval==0.4.8; extra == "harness"
+Provides-Extra: cuda
+Requires-Dist: flash-attn>=2.5.0; extra == "cuda"
+Provides-Extra: sparsify
+Requires-Dist: sparsify>=0.1.0; extra == "sparsify"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+<p align="center">
+  <img src="banner.png" alt="Wisent Banner" width="100%">
+</p>
+
+<p align="center">
+  <code>pip install wisent</code>
+</p>
+
+
+
+## Overview
+
+Wisent allows you to control your AI by identifying brain patterns corresponding to responses you don't like, like hallucinations or harmful outputs. We use contrastive pairs of representations to detect when a model might be generating harmful content or hallucinating. Learn more at [wisent.ai/documentation](https://www.wisent.ai/documentation).  
+
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

wisent-0.7.925/README.md

@@ -0,0 +1,18 @@
+<p align="center">
+  <img src="banner.png" alt="Wisent Banner" width="100%">
+</p>
+
+<p align="center">
+  <code>pip install wisent</code>
+</p>
+
+
+
+## Overview
+
+Wisent allows you to control your AI by identifying brain patterns corresponding to responses you don't like, like hallucinations or harmful outputs. We use contrastive pairs of representations to detect when a model might be generating harmful content or hallucinating. Learn more at [wisent.ai/documentation](https://www.wisent.ai/documentation).  
+
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

wisent-0.7.925/pyproject.toml

@@ -0,0 +1,85 @@
+[tool.pytest.ini_options]
+testpaths = ["wisent/tests"]
+python_files = ["test_*.py"]
+python_classes = ["Test*"]
+python_functions = ["test_*"]
+addopts = ["-v", "--tb=short"]
+markers = [
+    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
+    "heavy: marks tests as very resource-intensive with large model downloads (deselect with '-m \"not heavy\"')",
+    "model_download: marks tests that download models from HuggingFace (deselect with '-m \"not model_download\"')",
+    "integration: marks tests as integration tests",
+    "unit: marks tests as unit tests",
+    "docker: marks tests for Docker-based execution",
+    "performance: marks tests for performance evaluation",
+    "cli: marks tests for CLI integration",
+    "sandbox_required: marks tests that require sandbox environment (includes coding tasks and trust_remote_code)",
+    "bigcode_required: marks tests that require bigcode-evaluation-harness installation",
+]
+filterwarnings = [
+    "ignore::DeprecationWarning",
+    "ignore::PendingDeprecationWarning",
+]
+
+[tool.ruff]
+line-length = 120
+target-version = "py39"
+exclude = [
+    ".bzr",
+    ".direnv", 
+    ".eggs",
+    ".git",
+    ".git-rewrite",
+    ".hg",
+    ".mypy_cache",
+    ".nox",
+    ".pants.d",
+    ".pytype",
+    ".ruff_cache",
+    ".svn",
+    ".tox",
+    ".venv",
+    "__pypackages__",
+    "_build",
+    "buck-out",
+    "build",
+    "dist",
+    "node_modules",
+    "venv",
+    "evaluation",
+    "bigcode-evaluation-harness",
+]
+
+[tool.ruff.lint]
+# Enable common rule sets
+select = ["E4", "E7", "E9", "F", "I", "N", "UP", "B", "A", "C4", "ISC", "ICN", "PIE", "T20", "PT", "Q", "RSE", "RET", "SLF", "SIM", "TCH", "ARG", "ERA", "PGH", "RUF"]
+ignore = [
+    "PLR0913", # Too many arguments in function definition
+    "PLR0912", # Too many branches
+    "PLR0915", # Too many statements
+    "C901",    # Function is too complex
+    "PLR2004", # Magic value used in comparison
+    "TRY003",  # Avoid specifying long messages outside exception class
+    "PLW2901", # Redefined loop variable
+    "B008",    # Do not perform function calls in argument defaults
+    "ARG002",  # Unused method argument
+    "PIE800",  # Unnecessary spread operator,
+    "T201" # print
+]
+fixable = ["ALL"]
+unfixable = []
+
+[tool.ruff.format]
+# Like Black, use double quotes for strings.
+quote-style = "double"
+# Like Black, indent with spaces, rather than tabs.
+indent-style = "space"
+# Like Black, respect magic trailing commas.
+skip-magic-trailing-comma = false
+# Like Black, automatically detect the appropriate line ending.
+line-ending = "auto"
+
+[tool.ruff.lint.isort]
+known-first-party = ["wisent"]
+force-single-line = false
+combine-as-imports = true

wisent-0.7.925/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

wisent-0.7.925/setup.py

@@ -0,0 +1,78 @@
+import os
+from setuptools import setup, find_packages
+
+# Read the contents of the README file
+with open("README.md", encoding="utf-8") as f:
+    long_description = f.read()
+
+# Read version from __init__.py
+with open(os.path.join("wisent", "__init__.py"), encoding="utf-8") as f:
+    for line in f:
+        if line.startswith("__version__"):
+            version = line.split("=")[1].strip().strip('"').strip("'")
+            break
+
+setup(
+    name="wisent",
+    version=version,
+    author="Lukasz Bartoszcze and the Wisent Team",
+    author_email="lukasz.bartoszcze@wisent.ai",  # Replace with your email
+    description="Monitor and influence AI Brains",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    url="https://github.com/wisent-ai/wisent",  # Replace with your GitHub repo
+    packages=find_packages(exclude=["patches", "patches.*"]),  # Exclude patches directory
+    include_package_data=True,
+    package_data={
+        "wisent": [
+            "core/evaluators/benchmark_specific/coding/safe_docker/Dockerfile",
+            "core/evaluators/benchmark_specific/coding/safe_docker/entrypoint.py",
+            "parameters/lm_eval/*.json",
+            "parameters/tasks/*.json",
+            "examples/scripts/*.json",
+            "scripts/*.sh",
+        ],
+    },
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
+        "Development Status :: 4 - Beta",
+        "Intended Audience :: Science/Research",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    ],
+    python_requires=">=3.8",
+    install_requires=[
+        "torch>=1.9.0",
+        "transformers>=4.20.0",
+        "tqdm>=4.50.0",
+        "scikit-learn>=0.24.0",
+        "pandas>=1.2.0",
+        "numpy>=1.21.0",
+        "datasets>=2.0.0",
+        "sentence-transformers>=2.0.0",
+        "faiss-cpu>=1.7.0",
+        "uncensorbench>=0.2.0",
+        "pebble>=5.0.0",
+        "latex2sympy2_extended>=1.0.0",
+        "sae_lens>=0.1.0",
+        "trl>=0.7.0",
+    ],
+    extras_require={
+        "harness": [
+            "lm-eval==0.4.8",
+        ],
+        "cuda": [
+            "flash-attn>=2.5.0",
+        ],
+        "sparsify": [
+            "sparsify>=0.1.0",
+        ],
+    },
+    entry_points={
+        "console_scripts": [
+            "wisent=wisent.core.main:main",
+        ],
+    },
+    keywords="nlp, machine learning, language models, safety, guardrails, lm-evaluation-harness",
+)