
import os

from openfisca_core.taxbenefitsystems import TaxBenefitSystem

from . import entities


COUNTRY_DIR = os.path.dirname(os.path.abspath(__file__))


# Our country tax and benefit class inherits from the general TaxBenefitSystem class.
# The name CountryTaxBenefitSystem must not be changed, as all tools of the OpenFisca ecosystem expect a CountryTaxBenefitSystem class to be exposed in the __init__ module of a country package.
class CountryTaxBenefitSystem(TaxBenefitSystem):
    def __init__(self):
        # We initialize our tax and benefit system with the general constructor
        super(CountryTaxBenefitSystem, self).__init__(entities.entities)

        # We add to our tax and benefit system all the variables
        self.add_variables_from_directory(os.path.join(COUNTRY_DIR, 'variables'))

        # We add to our tax and benefit system all the legislation parameters defined in the  parameters files
        param_files = [
            '__root__.xml',
            'benefits.xml',
            'general.xml',
            'taxes.xml',
            ]

        for param_file in param_files:
            param_path = os.path.join(COUNTRY_DIR, 'parameters', param_file)
            self.add_legislation_params(param_path)


import os

from openfisca_core.taxbenefitsystems import TaxBenefitSystem

from . import entities


COUNTRY_DIR = os.path.dirname(os.path.abspath(__file__))


# Our country tax and benefit class inherits from the general TaxBenefitSystem class.
# The name CountryTaxBenefitSystem must not be changed, as all tools of the OpenFisca ecosystem expect a CountryTaxBenefitSystem class to be exposed in the __init__ module of a country package.
class CountryTaxBenefitSystem(TaxBenefitSystem):
    def __init__(self):
        # We initialize our tax and benefit system with the general constructor
        super(CountryTaxBenefitSystem, self).__init__(entities.entities)

        # We add to our tax and benefit system all the variables
        self.add_variables_from_directory(os.path.join(COUNTRY_DIR, 'variables'))

        # We add to our tax and benefit system all the legislation parameters defined in the  parameters files
        param_path = os.path.join(COUNTRY_DIR, 'parameters')
        self.add_legislation_params(param_path)

Use YAML params instead of XML params
Use YAML params instead of XML params

  
  import os
  
  from openfisca_core.taxbenefitsystems import TaxBenefitSystem
  
  from . import entities
  
  
  COUNTRY_DIR = os.path.dirname(os.path.abspath(__file__))
  
  
  # Our country tax and benefit class inherits from the general TaxBenefitSystem class.
  # The name CountryTaxBenefitSystem must not be changed, as all tools of the OpenFisca ecosystem expect a CountryTaxBenefitSystem class to be exposed in the __init__ module of a country package.
  class CountryTaxBenefitSystem(TaxBenefitSystem):
      def __init__(self):
          # We initialize our tax and benefit system with the general constructor
          super(CountryTaxBenefitSystem, self).__init__(entities.entities)
  
          # We add to our tax and benefit system all the variables
          self.add_variables_from_directory(os.path.join(COUNTRY_DIR, 'variables'))
  
          # We add to our tax and benefit system all the legislation parameters defined in the  parameters files
+         param_path = os.path.join(COUNTRY_DIR, 'parameters')
+         self.add_legislation_params(param_path)
-         param_files = [
-             '__root__.xml',
-             'benefits.xml',
-             'general.xml',
-             'taxes.xml',
-             ]
  
-         for param_file in param_files:
-             param_path = os.path.join(COUNTRY_DIR, 'parameters', param_file)
-             self.add_legislation_params(param_path)
- 
Use YAML params instead of XML params
## Code Before:

import os

from openfisca_core.taxbenefitsystems import TaxBenefitSystem

from . import entities


COUNTRY_DIR = os.path.dirname(os.path.abspath(__file__))


# Our country tax and benefit class inherits from the general TaxBenefitSystem class.
# The name CountryTaxBenefitSystem must not be changed, as all tools of the OpenFisca ecosystem expect a CountryTaxBenefitSystem class to be exposed in the __init__ module of a country package.
class CountryTaxBenefitSystem(TaxBenefitSystem):
    def __init__(self):
        # We initialize our tax and benefit system with the general constructor
        super(CountryTaxBenefitSystem, self).__init__(entities.entities)

        # We add to our tax and benefit system all the variables
        self.add_variables_from_directory(os.path.join(COUNTRY_DIR, 'variables'))

        # We add to our tax and benefit system all the legislation parameters defined in the  parameters files
        param_files = [
            '__root__.xml',
            'benefits.xml',
            'general.xml',
            'taxes.xml',
            ]

        for param_file in param_files:
            param_path = os.path.join(COUNTRY_DIR, 'parameters', param_file)
            self.add_legislation_params(param_path)

## Instruction:
Use YAML params instead of XML params
## Code After:

import os

from openfisca_core.taxbenefitsystems import TaxBenefitSystem

from . import entities


COUNTRY_DIR = os.path.dirname(os.path.abspath(__file__))


# Our country tax and benefit class inherits from the general TaxBenefitSystem class.
# The name CountryTaxBenefitSystem must not be changed, as all tools of the OpenFisca ecosystem expect a CountryTaxBenefitSystem class to be exposed in the __init__ module of a country package.
class CountryTaxBenefitSystem(TaxBenefitSystem):
    def __init__(self):
        # We initialize our tax and benefit system with the general constructor
        super(CountryTaxBenefitSystem, self).__init__(entities.entities)

        # We add to our tax and benefit system all the variables
        self.add_variables_from_directory(os.path.join(COUNTRY_DIR, 'variables'))

        # We add to our tax and benefit system all the legislation parameters defined in the  parameters files
        param_path = os.path.join(COUNTRY_DIR, 'parameters')
        self.add_legislation_params(param_path)
