Bump torch and ax versions
from setuptools import setup
setup(
- name="kge",
+ name="libkge",
version="0.1",
description="A knowledge graph embedding library",
url="https://github.com/rufex2001/kge",
@@ -9,12 +9,12 @@ setup(
author_email="rgemulla@uni-mannheim.de",
packages=["kge"],
install_requires=[
- "torch>=1.3.0",
+ "torch>=1.3.1",
"pyyaml",
"pandas",
"argparse",
"path.py",
- "ax-platform>=0.1.2",
+ "ax-platform>=0.1.6",
"sqlalchemy",
"torchviz",
"dataclasses"
