def input_schema(self):
     return sb.PartialSchema('bun', [
         sb.num('bun')
     ])
 def input_schema(self):
     return sb.PartialSchema('age', [
         sb.num('age_at_admission')
     ])