Add entrypoints for the composites and samplers

dwavesystems · Dec 27, 2018 · 23d60d8 · 23d60d8
1 parent b0469a2
commit 23d60d8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -66,6 +66,14 @@
     'Programming Language :: Python :: 3.7',
     ]
 
+entry_points = {'dwave.samplers': ['ExactSolver = dimod:ExactSolver',
+                                   'RandomSampler = dimod:RandomSampler',
+                                   ],
+                'dwave.composites': ['StructureComposite = dimod:StructureComposite',
+                                     'SpinReversalTransformComposite = dimod:SpinReversalTransformComposite',
+                                     'FixedVariableComposite = dimod:FixedVariableComposite',
+                                     ]}
+
 python_requires = '>=2.7,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*'
 
 try:
@@ -158,6 +166,7 @@ def run_setup(cpp):
         classifiers=classifiers,
         zip_safe=False,
         python_requires=python_requires,
+        entry_points=entry_points,
         **kw
     )