class DsContext(object):
    def __init__(self, config="okdataset.yml"):
        self.workers = 8

import yaml
import os

"""
DataSet context
"""
class Context(object):
    def __init__(self, config=os.path.dirname(os.path.realpath(__file__)) + "/../okdataset.yml"):
        self.workers = 8
        self.config = yaml.load(open(config).read())


Put yaml config in Context.
Put yaml config in Context.

+ import yaml
+ import os
+ 
+ """
+ DataSet context
+ """
- class DsContext(object):
+ class Context(object):
-     def __init__(self, config="okdataset.yml"):
+     def __init__(self, config=os.path.dirname(os.path.realpath(__file__)) + "/../okdataset.yml"):
          self.workers = 8
+         self.config = yaml.load(open(config).read())
  
+ 
Put yaml config in Context.
## Code Before:
class DsContext(object):
    def __init__(self, config="okdataset.yml"):
        self.workers = 8

## Instruction:
Put yaml config in Context.
## Code After:
import yaml
import os

"""
DataSet context
"""
class Context(object):
    def __init__(self, config=os.path.dirname(os.path.realpath(__file__)) + "/../okdataset.yml"):
        self.workers = 8
        self.config = yaml.load(open(config).read())

