defined_as: kipoi.model.PyTorchModel
args:
    module_file: model_files/pyt.py
    module_obj: simple_model
    weights: model_files/only_weights.pth
default_dataloader: . # path to the directory
info:
    authors:
        - name: Roman Kreuzhuber
          github: krrome
    doc: Simple testing model for pytorch
dependencies:
    conda:
        - pytorch::pytorch>=0.2.0
schema:
    inputs:
        name: input
        shape: (1000, 4)
        special_type: DNASeq
        doc: DNA sequence
        # associated_metadata: ranges # --> has to be defined in dataloader.yaml.
        #                                   This field is ignored in model.yaml.
    targets:
        shape: (1, )
        doc: Predicted binding strength
        column_labels:
            - some_probability
test:
     expect:
       url: https://s3.eu-central-1.amazonaws.com/kipoi-models/predictions/example/models/pyt/expected.pred.h5
       md5: d6d0779a7bdfb1301c76a59defd293ed
     precision_decimal: 6