Skip to content
Snippets Groups Projects
Commit 08353704 authored by David Hammer's avatar David Hammer
Browse files

Arbiter kernels call super().__init__

parent d113cd10
No related branches found
No related tags found
1 merge request!53Train picker arbiter kernel
...@@ -8,6 +8,7 @@ from .base_kernel import BaseArbiterKernel ...@@ -8,6 +8,7 @@ from .base_kernel import BaseArbiterKernel
class BooleanCombination(BaseArbiterKernel): class BooleanCombination(BaseArbiterKernel):
def __init__(self, config): def __init__(self, config):
super().__init__(config)
self._key = config.get("key") self._key = config.get("key")
self._operator = getattr(np, config.get("operator")) self._operator = getattr(np, config.get("operator"))
......
...@@ -12,6 +12,7 @@ class PpuKernel(BaseArbiterKernel): ...@@ -12,6 +12,7 @@ class PpuKernel(BaseArbiterKernel):
_node_name = "ppu" _node_name = "ppu"
def __init__(self, config): def __init__(self, config):
super().__init__(config)
self._ppu_device_id = config.get("ppuDevice") self._ppu_device_id = config.get("ppuDevice")
@staticmethod @staticmethod
......
...@@ -9,8 +9,7 @@ from .base_kernel import BaseArbiterKernel ...@@ -9,8 +9,7 @@ from .base_kernel import BaseArbiterKernel
class RandomSampler(BaseArbiterKernel): class RandomSampler(BaseArbiterKernel):
def __init__(self, config): def __init__(self, config):
print("Getting config") super().__init__(config)
print(config)
self._threshold = (100 - config.get("probability")) / 100 self._threshold = (100 - config.get("probability")) / 100
@staticmethod @staticmethod
......
...@@ -11,6 +11,7 @@ from .base_kernel import BaseArbiterKernel ...@@ -11,6 +11,7 @@ from .base_kernel import BaseArbiterKernel
class ReduceAndThreshold(BaseArbiterKernel): class ReduceAndThreshold(BaseArbiterKernel):
def __init__(self, config): def __init__(self, config):
super().__init__(config)
self._key = config.get("key") self._key = config.get("key")
self._threshold = config.get("threshold") self._threshold = config.get("threshold")
self._reduction = getattr(np, config.get("reduction")) self._reduction = getattr(np, config.get("reduction"))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment