Convert CodeSearchResult to ShortRepository

sigmavirus24 · sigmavirus24 · commit d6b80a9eacb5 · 2017-02-06T20:44:49.000-06:00
diff --git a/github3/search/code.py b/github3/search/code.py
@@ -1,11 +1,11 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
 
-from ..models import GitHubCore
-from ..repos import Repository
+from .. import models
+from .. import repos
 
 
-class CodeSearchResult(GitHubCore):
+class CodeSearchResult(models.GitHubCore):
 
     def _update_attributes(self, data):
         self._api = self._get_attribute(data, 'url')
@@ -27,7 +27,7 @@ def _update_attributes(self, data):
 
         #: Repository the code snippet belongs to
         self.repository = self._class_attribute(
-            data, 'repository', Repository, self
+            data, 'repository', repos.ShortRepository, self
         )
 
         #: Score of the result
