from django.contrib import admin

from .models import YelpLocation, YelpReview


class YelpLocationAdmin(admin.ModelAdmin):
    list_display = ['location_id', 'url']


admin.site.register(YelpLocation, YelpLocationAdmin)
admin.site.register(YelpReview)