<commit_before>#!/usr/bin/env python3
from choo.networks import VRR
from choo.models import Serializable, Stop, Location, Trip, unserialize_typed
import json

# collection = Collection('test')

vrr = VRR()
bs = Stop(city='essen', name='fliegenbusch')
bo = Stop(city='essen', name='hbf')

trip = Trip.Request()
trip.origin = bs
trip.destination = bo

location = Location.Request()
location.name = 'Borbeck'


# result = vrr.search_trips(trip)

unserialize_typed

result = vrr.query(bs)

# import pprint
# pp = pprint.PrettyPrinter(indent=1)
# pp.pprint(result.serialize())<commit_msg>test.py<commit_after>#!/usr/bin/env python3
from choo.networks import VRR
from choo.models import Serializable, Stop, Location, Trip
import json

# collection = Collection('test')

vrr = VRR()
bs = Stop(city='essen', name='<NAME>')
bo = Stop(city='essen', name='hbf')
# bo = Stop(city='heidelberg', name='hbf')

trip = Trip.Request()
trip.origin = bs
trip.destination = bo

location = Location.Request()
location.name = 'Borbeck'


result = vrr.query(bs)

# import pprint
# pp = pprint.PrettyPrinter(indent=1)
# pp.pprint(result.serialize())
{'fraction_non_alphanumeric': 0.15302491103202848, 'fraction_numerical': 0.0035587188612099642, 'mean_word_length': 5.617647058823529, 'pattern_counts': {'":': 0, '<': 4, '<?xml version=': 0, '>': 4, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '2505592', 'n_tokens_mistral': 415, 'n_tokens_neox': 392, 'n_words': 90}